在 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: 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")