Keras-inference-time-optimizer icon indicating copy to clipboard operation
Keras-inference-time-optimizer copied to clipboard

Published 20 hours ago • verified publisher icon ZFTurbo

add support for tensorflow.keras API

Open evdem opened this issue 5 years ago • 2 comments

I get "TypeError: init() got an unexpected keyword argument 'ragged'" for model = tf.keras.applications.MobileNet(input_shape=IMG_SHAPE, include_top=True, weights='imagenet') model_reduced = reduce_keras_model(model)

See https://stackoverflow.com/questions/58878421/unexpected-keyword-argument-ragged-in-keras "...model you have exported is from tensorflow.keras and not directly from keras API. These two are different. So while loading it might be using tf.ragged tensors that might not be compatible with keras API...". They suggest change 'from keras...' to 'from tensorflow.keras...'

evdem avatar Dec 27 '19 12:12 evdem

Any updates about this?

Choapinus avatar Jun 25 '20 23:06 Choapinus

I made support for tensorflow.keras. It must work fine now.

ZFTurbo avatar Jul 18 '20 09:07 ZFTurbo