Сохраните контрольную точку 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 не существует

Я выполнил предложенный ответ здесь ,

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

Символ "/" обозначает "дом" Колаба.Предполагая, что ваш персональный диск смонтирован как:

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