tensorflow - 在 Google Colab 中保存 TensorFlow 检查点

Question

我正在尝试使用 TF.train.Saver() 保存我的训练模型

saver = tf.train.Saver()

with tf.Session() as sess:
  [...]
  save_path = saver.save(sess, "/final_model.ckpt")

我知道这可以在本地计算机上运行，​​但是当我尝试在 Google Colab 上运行它时，它不会运行，并且会出现以下错误:

ValueError: Parent directory of final_model.ckpt doesn't exist

我已经执行了建议的答案 here ,

from google.colab import drive
drive.mount('/content/gdrive/')

但是我保存的模型不会放在当前目录下。它进入我在“我的云端硬盘”中看不到的 content 文件夹。

我的代码在 My Drive/TF 01/some method/ 下，我想将检查点和日志保存在同一个文件夹中。

Answer 1

这是因为行:

save_path = saver.save(sess, "/final_model.ckpt")

“/”符号指的是 Colab 的“家”。假设您的个人驱动器安装为:

from google.colab import drive
drive.mount('/content/gdrive/')

您可以通过这种方式保存在您的驱动器中:

save_path = saver.save(sess, "gdrive/My Drive/TF 01/some method/final_model.ckpt")

或者通过硬编码:

save_path = saver.save(sess, "/content/gdrive/My Drive/TF 01/some method/final_model.ckpt")

或者，如果您只想在运行时使用它，只需删除“/”符号:

save_path = saver.save(sess, "final_model.ckpt")