我该如何在Google Colab上存储我的训练模型并在本地磁盘上检索?使用检查点可以吗?我该如何存储它们并在一段时间后检索它们?请提供相关代码。谢谢。
我该如何在Google Colab上存储我的训练模型并在本地磁盘上检索?使用检查点可以吗?我该如何存储它们并在一段时间后检索它们?请提供相关代码。谢谢。
当您打开笔记本时,会创建Google Colab实例,稍后会被删除,因此您无法在不同的运行中访问数据。如果您想将训练好的模型下载到本地计算机上,则可以使用以下方法:
from google.colab import files
files.download(<filename>)
如果你想从本地上传模型,可以执行以下操作:
from google.colab import files
files.upload(<filename>)
另一种可能(并且在我看来更好的)解决方案是使用Github存储您的模型,只需提交和推送您的模型到Github,稍后克隆该存储库即可获取模型。
> import os
> checkpoint_path = "training_1\cp.ckpt"
> checkpoint_dir = os.path.dirname(checkpoint_path)
# Create checkpoint callback
> cp_callback =ModelCheckpoint(checkpoint_path,
monitor='val_acc',save_best_only=True,save_weights_only=True,verbose=1)
> network_fit = myModel.fit(x, y, batch_size=25, epochs=20,
,callbacks = [cp_callback] )
myModel.load_weights(checkpoint_path)