如何将张量流模型保存到谷歌驱动器

Question

我在 google colab 中使用张量流，我想将我的模型保存在 google 驱动器上以供将来使用，有人可以指导我如何做到这一点吗？

Answer 1

首先连接到谷歌驱动器：

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

接下来将您的模型保存在 colab 上

model.save('modelname')

编辑：我们还可以指定保存模型的路径（在谷歌驱动器路径中对我来说效果很好），如下所示。该代码还以 h5 格式保存模型以避免警告，并且模型在加载时可以完全重用：

model.save('/content/drive/MyDrive/MyFolder/myModel', save_format="h5")

保存的模型现在将显示在“内容”文件夹中。现在，在 colab 的“gdrive”文件夹中创建相同的文件夹以及子文件夹。您可以使用shutil 将相关文件复制到这些文件夹中。

import shutil
shutil.copy('source','destination')

您可以使用“复制路径”选项来获取文件的确切地址。最后要加载模型，您必须连接到驱动器并加载模型。

model = tf.keras.models.load_model('modeladdress')