在Google Colab上,我们可以直接训练深度学习模型,接下来介绍一下Pytorch如何保存和恢复模型。
首先,将你的Google Drive挂载上。
-
点击Mount Drive
-
运行这段代码(上一步完成后,会自动生成的)
-
在文件夹中就能看到已挂载的Drive文件夹
保存模型
PATH = './drive/My Drive/tran.pth'
torch.save(best_model.state_dict()
在Google Colab上,我们可以直接训练深度学习模型,接下来介绍一下Pytorch如何保存和恢复模型。
首先,将你的Google Drive挂载上。
点击Mount Drive
运行这段代码(上一步完成后,会自动生成的)
在文件夹中就能看到已挂载的Drive文件夹
保存模型
PATH = './drive/My Drive/tran.pth'
torch.save(best_model.state_dict()