-
Notifications
You must be signed in to change notification settings - Fork 47
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Errow while save and load the model #20
Comments
Hi @sachinkmohan , so the problem you describe requires a little larger fix, which I due to a lack of time will not be able to fix in the near term. |
Thanks for the response! @JanMarcelKezmann Appreciate it! |
Hi, I am having similar issue but with saving of the weights. Just had to redo anaconda setup and reinstall tensorflow. model.save(os.path.join(new_folder, 'model_weights.h5')) I am now getting this: C:\Users\Master\anaconda3\envs\gputensorflow\lib\site-packages\tensorflow\python\keras\utils\generic_utils.py:494: CustomMaskWarning: Custom mask layers require a config and must override get_config. When loading, the custom mask layer must be passed to the custom_objects argument. Python 3.10.7 (tags/v3.10.7:6cc6b13, Sep 5 2022, 14:08:36) [MSC v.1933 64 bit (AMD64)] on win32
Appreciate your insight what must i do next? |
I had uncommented the checkpoint and tried to save the model. It generated
saved_model.pb
file. Below are the warning messages when the model was saved.Now when loading the model, I am getting the following error.
model = tf.keras.models.load_model('/home/mohan/git/TensorFlow-Advanced-Segmentation-Models/examples/DeepLabV3plus1.ckpt/')
I am not sure how to load the custom objects though. Please help.
The text was updated successfully, but these errors were encountered: