Tensorflow模型保存与加载

1. 基本代码和模型文件介绍 1.1 保存模型 saver = tf.train.Saver(max_to_keep=5) saver.save(sess, 'model/my_model.ckpt', global_step=epoch) 1.2 模型文件 checkpoint 最新的模型文件信息 data和index 训练好的权重、偏置等数据 meta 图结构,包括变量、操作、集合等 1.3
相关文章
相关标签/搜索