tensorflow 保存模型

1
2
3
4
5
|--checkpoint_dir
| |--checkpoint
| |--MyModel.meta
| |--MyModel.data-00000-of-00001
| |--MyModel.index

meta文件

MyModel.meta文件保存的是图结构,meta文件是pb(protocol buffer)格式文件,包含变量、op、集合等。

ckpt文件

ckpt文件是二进制文件,保存了所有的weights、biases、gradients等变量

1
2
MyModel.data-00000-of-00001
MyModel.index

save

1
saver.save(sess, './checkpoint_dir/MyModel',global_step=step,write_meta_graph=False)

load

1
2
new_saver = tf.train.import_meta_graph('./checkpoint_dir/MyModel-1000.meta')
new_saver.restore(sess, tf.train.latest_checkpoint('./checkpoint_dir'))

https://blog.csdn.net/huachao1001/article/details/78501928

请作者喝一杯咖啡☕️