ValueError: Unknown layer: TCN when trying to load saved model

[soso-maitha]

Hello,

I get the following error after loading the model. I don't get this error with other saved models. Kindly advise as I have sumbssion shortly and I took more than 5 hours to train a complex model I don't have time.

Thank you very much in advance

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
[<ipython-input-15-acb631543b3f>](https://localhost:8080/#) in <module>()
      1 from tensorflow.keras.models import load_model
----> 2 savedModel=load_model("drive/MyDrive/Colab Notebooks/models/tcn_model7.h5")
      3 #savedModel.summary()

1 frames
[/usr/local/lib/python3.7/dist-packages/keras/utils/generic_utils.py](https://localhost:8080/#) in class_and_config_for_serialized_keras_object(config, module_objects, custom_objects, printable_module_name)
    561   if cls is None:
    562     raise ValueError(
--> 563         f'Unknown {printable_module_name}: {class_name}. Please ensure this '
    564         'object is passed to the `custom_objects` argument. See '
    565         'https://www.tensorflow.org/guide/keras/save_and_serialize'

ValueError: Unknown layer: TCN. Please ensure this object is passed to the `custom_objects` argument. See https://www.tensorflow.org/guide/keras/save_and_serialize#registering_the_custom_object for details.

[psomers3]

Take a look how this is done here: https://github.com/philipperemy/keras-tcn/blob/master/tasks/save_reload_sequential_model.py

[soso-maitha]

thanks a lot for your prompt support it is working. i will reference your code and paper in my report. Great work!

[philipperemy]

@psomers3 Thanks a lot for helping out!