1 | |--checkpoint_dir |
meta文件
MyModel.meta文件保存的是图结构,meta文件是pb(protocol buffer)格式文件,包含变量、op、集合等。
ckpt文件
ckpt文件是二进制文件,保存了所有的weights、biases、gradients等变量
1 | MyModel.data-00000-of-00001 |
save
1 | saver.save(sess, './checkpoint_dir/MyModel',global_step=step,write_meta_graph=False) |
load
1 | new_saver = tf.train.import_meta_graph('./checkpoint_dir/MyModel-1000.meta') |