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

Support of Tensorflow 2.x backend Keras models

Open rose-jinyang opened this issue 5 years ago • 3 comments

Hello How are you? Thanks for contributing this project. It seems that this project does not support Tensorflow 2.x backend Keras models. Could u support this? Thanks

rose-jinyang avatar Sep 08 '20 05:09 rose-jinyang

I made support for tensorflow.keras several weeks ago. It must work fine now. If you have problem please post error message.

ZFTurbo avatar Sep 08 '20 07:09 ZFTurbo

Hello there, me again.

I got this error after I excecuted

from kito import reduce_keras_model model_reduced = reduce_keras_model(model)

image

tensorflow version: 1.15.0 keras version: 2.2.4-tf

Best regards!

Choapinus avatar Sep 11 '20 19:09 Choapinus

@Choapinus do you have some custom layers in your model?

Also check this: https://stackoverflow.com/questions/58878421/unexpected-keyword-argument-ragged-in-keras

ZFTurbo avatar Sep 12 '20 21:09 ZFTurbo