philipperemy/keras-tcn

the model save but dosnt load - ValueError: Unknown layer: TCN

atila-gh opened this issue · 4 comments

atila-gh commented

thanks , it is very fast and has very good Performance

Unfortunately - i can save Model but cant load it again , always i encounter with below error!!!!!
ValueError: Unknown layer: TCN

please Help
for example : your sample code
001

philipperemy commented

@nas2000co can you follow the example in

keras-tcn/tasks/save_reload_sequential_model.py

Line 1 in 23845be

import numpy as np
and tell me?

  • 👍1
  • 😄1
  • 🎉1
philipperemy commented

I guess you need to give custom_objects={'TCN': TCN} when you load the model. Otherwise the layer will be unknown.

  • 👍1
  • 😄1
  • 🎉1
atila-gh commented

Thanks So much thats worked very well !!!!!!!!!!!!!!!!!!!!

new_model= tf.keras.models.load_model('model.h5' , custom_objects={'TCN': TCN})

philipperemy commented

Great!!!! I'll close this issue. Happy I could help :)