Google Colab 保存和恢复模型（Pytorch）

在Google Colab上，我们可以直接训练深度学习模型，接下来介绍一下Pytorch如何保存和恢复模型。

首先，将你的Google Drive挂载上。

1. 点击Mount Drive
   

2. 运行这段代码（上一步完成后，会自动生成的）
   

3. 在文件夹中就能看到已挂载的Drive文件夹
   

保存模型

PATH = './drive/My Drive/tran.pth'
torch.save(best_model.state_dict(), PATH)

恢复模型（保证前面挂载了Google Drive的前提下，进行恢复）

PATH = './drive/My Drive/tran.pth'
checkpoint = torch.load(PATH)
model.load_state_dict(checkpoint)

问题：

1. 为什么要把模型保存到Google Drive？
   由于之前Google Drive运行一段时间，人不进行操作或网络不稳定，服务器会自动停止运行。保存到Google Drive更放心一些。