You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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
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)
The text was updated successfully, but these errors were encountered: