Pretrained Model: is it still there? if so, how to "tf.load_model" it?
GastonMazzei opened this issue · 0 comments
GastonMazzei commented
Hello,
The README.md points to this GDrive link to fetch the pretrained model, but it seems to have only the following files. Is this correct?
checkpoint
epoch_200.ckpt.index
epoch_200.ckpt.data-00000-of-00001
epoch_200.ckpt.meta
If so, can I kindly ask if it is possible to load it in tensorflow (plug-and-play) and how to do it?
e.g.
model = tf.keras.models.load_model('trained_models/epoch_200.ckpt.data-00000-of-00001')
yields
>OSError: SavedModel file does not exist at: trained_models/epoch_200.ckpt.data-00000-of-00001/{saved_model.pbtxt|saved_model.pb}
Thank you, Gaston.