当我使用 Google Colaboratory 时，如何在我的 Google 云端硬盘中保存图像、权重？

Question

我使用 Google Colaboratory，然后我想将输出图像保存在我的 Google Drive 或 SSD、HHD 中，但它的目录是“/content”

 import os
print(os.getcwd())
# "/content"

那么是否可以更改路径（HDD、SSD、googledrive）？

原文由 이성령 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

为了节省重量，您可以在训练后运行以下内容。

 saver = tf.train.Saver()
save_path = saver.save(session, "data/dm.ckpt")
print('done saving at',save_path)

检查保存 ckpt 文件的位置。

 import os
print( os.getcwd() )
print( os.listdir('data') )

终于下载文件了！

 from google.colab import files
files.download( "data/dm.ckpt.meta" )

原文由 Nazmus Sakib 发布，翻译遵循 CC BY-SA 3.0 许可协议

Answer 2

您需要将 google drive 安装到您的 Colab 会话中。

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

然后，您可以像写入本地文件系统一样简单地写入谷歌驱动器，如下所示：

 with open('/content/gdrive/My Drive/file.txt', 'w') as f:
  f.write('content')

原文由 Tadej Magajna 发布，翻译遵循 CC BY-SA 4.0 许可协议