如何将 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')
我在 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')