Skip to content
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

Unsuccessful TensorSliceReader constructor: Failed to find any matching files for #3

Open
mazzasaverio opened this issue Feb 19, 2022 · 1 comment

Comments

@mazzasaverio
Copy link

mazzasaverio commented Feb 19, 2022

Ii is not possible lead saved_model.pb because it is trained with TPU and I think it is necessary to set experimental_io_device, otherwise error occur trying to find temporary folder where to find variables

model_dir = "./model"

localhost_save_option = tf.saved_model.SaveOptions(experimental_io_device="/job:localhost")
model.save(model_dir, options=localhost_save_option)

Restore the weights
model2 = tf.keras.models.load_model(model_dir, options=localhost_save_option)

@rubensanchezrivero
Copy link

is there any update on this?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants