search

YingZhangDUT / Cross-Modal-Projection-Learning

TensorFlow Implementation of Deep Cross-Modal Projection Learning
MIT License
94 stars 20 forks source link

Loading ResNet-v1-152 #8

Closed davidstap closed 5 years ago

davidstap commented 5 years ago

I am unable to load the ResNet-v1-152 checkpoint, which I downloaded at the location this repo suggests.

When I give only the path I get the following message: No checkpoint Found, Please provide pretrained checkpoint (It is easy to miss this, since the model simply trains from scratch, i.e. without pre-trained weights when this happens.)

When I force the model to load the .ckpt file (by changing the line in configuration.py to ckpt = tf.train.get_checkpoint_state(FLAGS.restore_path, latest_filename='rennet_v1_152.ckpt')), I get the following message:

WARNING:tensorflow:ParseError: 1:1 : '???: Expected identifier or number, got . WARNING:tensorflow:/$PATH: checkpoint ignored No checkpoint Found, Please provide pretrained checkpoint

I have tried different Tensorflow versions, all with the same result. How do I load the .ckpt file?

davidstap commented 5 years ago

I solved the issue. Simply downloading the checkpoint and providing the path is not sufficient. In addition, in the checkpoint folder create a file called checkpoint and add the following two lines to the file:

model_checkopint_path: "resnet_v1_152.ckpt" all_model_checkpoint_paths: "resnet_v1_152.ckpt"

Now, loading the checkpoint should work.