如何保存我训练过的神经网络？

Question

我刚刚在https://colab.research.google.com的ipynb上完成了我的神经训练。

我能够从Google云端硬盘加载csv文件，但不知道如何在训练后保存我的体重矩阵。你们有个主意吗？

Answer 1

您可以在培训后运行以下命令。

saver = tf.train.Saver()
save_path = saver.save(session, "data/dm.ckpt")
print('done saving at',save_path)

然后检查保存ckpt文件的位置。

import os
print( os.getcwd() )
print( os.listdir('data') )

最后下载文件！

from google.colab import files
files.download( "data/dm.ckpt.meta" )