laughtervv/SGPN

Pretrained Model: is it still there? if so, how to "tf.load_model" it?

GastonMazzei opened this issue · 0 comments

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.