如何将 tensorflow 模型保存到 google 驱动器

How to save tensorflow model to google drive

我在 google colab 中使用张量流,我想将我的模型保存在 google 驱动器上以备将来使用,谁能指导我如何操作?

首先连接到 google 驱动器:

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

接下来将您的模型保存到 colab

model.save('modelname')

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

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

您可以使用 'copy path' 选项来获取文件的确切地址。最后要加载您的模型,您必须连接到驱动器并加载模型。

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