将TensorFlow检查点保存在Google Colab中

Question

我正在尝试使用TF.train.Saver()

保存训练有素的模型

saver = tf.train.Saver()

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

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

  

ValueError：final_model.ckpt的父目录不存在

我已经执行了建议的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")