mimic
mimic

Reputation: 5224

Tensorflow - which is the real checkpoint file?

I'm trying to used pretrained models with checkpoints but I can't figure out which file should be put into config as checkpoint. There are files:

I tried all of them but I see errors. In the different articles I saw just "model.ckpt" but there is no such files.

I tried to work with ssd_mobilenet from here: https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md

Upvotes: 4

Views: 4153

Answers (1)

Junyeop Lee
Junyeop Lee

Reputation: 247

  • model.ckpt.meta : File to store graph information
  • model.ckpt.index : File to store index of variables
  • model.ckpt.data0000-of-0001 : File to store value of variables

All three files are real checkpoint files. When you restore the model using tf.train.Saver.restore, parameter "save_path" should be "~model.ckpt". Your error probably occurred because of the invalid save_path. Check if three files(.meta, .index, .data~) in save_path(relative path or absolute path).

Upvotes: 7

Related Questions