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
相關文章
相關標籤/搜索