-
.data-00000-of-00001和.index保存了所有的weights、biases、gradients等变量, .meta保存了图结构, checkpoint文件是个文本文件,里面记录了保存的最新的checkpoint文件以及其它checkpoint文件列表
-
保存训练模型:
-
建立一个Saver对象
-
设置保存模型的个数
:
-
默认为5,保存最近 5个 模型
-
也可以每训练一个epoch就保存一次模型
-
-
保存模型
-
-
模型恢复
-
-
You can add it/replace in the demo.py , just after the parsing of the args. Then give the path to an existing model file e.g :
-
python ./faster_rcnn/demo.py --model dir_model_path/model.ckpt.meta
-