Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Tensorflow - which is the real checkpoint file?

Tags:

tensorflow

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:

  • model.ckpt.meta
  • model.ckpt.index
  • model.ckpt.data0000-of-0001

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

like image 288
mimic Avatar asked Mar 08 '23 13:03

mimic


1 Answers

  • 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).

like image 148
Junyeop Lee Avatar answered Mar 15 '23 18:03

Junyeop Lee