保存-恢复模型(不需要重新定义模型的网络结构)-基于tensorflow1.2
以下内容基于TensorFlow1.2
checkpoints保存和恢复模型
保存模型
定义要输出的tensor
output_name = ...
# 如果是预测,output_tensor就是最后得出预测结果的tensor(预测的y)
output_tensor = ...
tf.add_to_collection(output_name, output_tensor)
保存checkpoint
checkpoint_dir = ...
global_step = ...
with t
原创
2020-07-16 17:43:10 ·
231 阅读 ·
0 评论