-
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
save as pb #22
Comments
according to the official documents, it could only be saved as h5 or tf format, how it could be saved as pb format |
Hey, I faced the same error when I tried to create the pb file in Google
Colab instead of my AWS environment, however it works just fine in my AWS
environment. I'm still trying to figure it out. I'll let you know once it's
sorted out. In the meantime, if you figure it out, please let me know.
Thanks for pointing it out!
…On Wed, Nov 4, 2020 at 3:25 AM zhenglongjiepheonix ***@***.***> wrote:
when I am trying to tf.keras.models.load_model('new_model.h5'), which is
gererated by save_model after loading your 'model.h5' ,it seems something
wrong with your lambda layer
[image: pb]
<https://user-images.githubusercontent.com/32992656/98086721-1b8e6800-1eba-11eb-97bc-22156cfde51b.png>
—
You are receiving this because you are subscribed to this thread.
Reply to this email directly, view it on GitHub
<#22 (comment)>,
or unsubscribe
<https://github.com/notifications/unsubscribe-auth/AB7IJ5RPVEQVP3YGIKLT7ULSOEFXBANCNFSM4TJUSLUA>
.
|
I even tried to convert h5 to pb but when I use model.save() to save the file as "model.h5" , and later try to load using load_model("model.h5"), some error occurs. I think it would be better to implement in TF 1.x framework instead of using high level API like keras, since TF 1.x seems to support pb file better |
After all pb file is more important for deployment and acceleration |
I think to load the model without using Keras, use |
Hai when i train the model i got the error |
hey, i am using model.save as you mentioned so that i could get .pb file, but it turns out that i only get a file without any suffixes which is not in pb format
The text was updated successfully, but these errors were encountered: