google-colaboratory - 如何保存训练有素的神经网络？

Question

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

我能够从 Google Drive 加载一个 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" )