利用Google Colab训练模型

1 向Colab授予权限，以便他可以访问他并安装驱动器

from google.colab import drive

drive.mount('/content/drive')

2.如何将模型保存在google drive上

model_save_name = 'classifier.pt'

path = F"/content/gdrive/My Drive/{model_save_name}"

torch.save(model.state_dict(), path)

3.如何调用我们保存的模型

model_save_name = 'classifier.pt'

path = F"/content/gdrive/My Drive/{model_save_name}"

model.load_state_dict(torch.load(path))