python - 如何将 tensorflow 模型保存到谷歌驱动器

Question

我在 google colab 中使用 tensorflow ，我想将我的模型保存在 google drive 上以备将来使用，谁能指导我如何操作？

Answer 1

首先连接到谷歌驱动器:

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

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

model.save('modelname')

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

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

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

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