No checkpoint file found

[ga84]

Hello! This is a fairly stupid issue but it brings a lot of trouble. Can you please help me specify the path to checkpoint file? I tried everything including absolute path and specifying name of ckpt... I first copied checkpoint files to a folder 'model' and that should work with default settings in code, however it does not... Suggestions?

@hellochick

[intvo]

did you find a solution to this?

[ga84]

Unfortunately not yet. Can anyone of you guys who probably got this done please help? @sainttelant @Leequens @PascalWB @anny123123

[MAdil45]

Same issue. Kindly let us know the solution. We really appreciate that. @hellochick

[jiaxinxie97]

PSPNet-tensorflow/inference.py

Line 103 in 989ac5f

ckpt = tf.train.get_checkpoint_state(args.checkpoints)

This function can't work in their dir. Changing ckpt.model_checkpoint_path to $MODEL_PATH/model.ckpt-0 will be ok.

[ga84]

@jiaxinxie97 thank you for answering! However, I'm afraid it still doesn't work. Did you mean, writing e.g.
./model/pspnet101/model.ckpt-0 instead of ckpt.model_checkpoint_path
? I tried this and still not found.

I also tried setting SNAPSHOT_DIR as './model/pspnet101/model.ckpt-0', also tried './model/pspnet101/' and I tried with absolute paths in all combinations but no success... Can you please tell me where I might be doing wrong? I'd really appreciate that. Thank you!

EDIT: sorry for bad formatting, I have to type from my phone...

[ga84]

I solved the problem, by deleting the condition:
if ckpt and ckpt.model_checkpoint_path
And also the "else" part. And then after inserting the path to model.ckpt-0 instead of the ckpt.model_checkpoint_path it is finally possible to load checkpoint files. Thank you for the hint @jiaxinxie97 !

[MAdil45]

I have solved this by providing the checkpoint path directly at default line.
parser.add_argument("--checkpoints", type=str, default='/model/ade20k_model/',

[Jason-xin]

I have solved this by modifying 'checkpoint.txt' to 'checkpoint'

[ga84]

Thanks guys, I think we can close this issue!

[simardeep1792]

Hi Guys, could you explain how you all did it. I repeated all the steps and still it shows checkpoint not found. Could you atleat paste the screenshot of the code. @hellochick