tf.train.Checkpoint：在Google Colab中保存张量流检查点

Question

从google.colab导入文件

我正在尝试使用以下代码从Google colab笔记本中保存检查点。 -drive.mount是否创建文件夹“ gdrive” -我什至创建了文件夹并测试了保存，但无法做到

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

checkpoint_dir = '/content/gdrive/'
checkpoint_prefix = os.path.join(checkpoint_dir, "ckpt")
checkpoint = tf.train.Checkpoint(optimizer=optimizer,
                                 encoder=encoder,
                                 decoder=decoder)