保存并重新加载先前训练的模型

Question

我正试图保存我的半训练模型并再次重新加载以进行进一步的训练，以避免再次从头开始训练它的问题。 我正在使用以下代码。

model_save_name = 'trained_model'
path = F"/content/drive/My Drive/{model_save_name}" 
torch.save(model.state_dict(), path)

但是它显示以下错误。

----> 5 torch.save(model.state_dict(), path)

AttributeError: 'Model' object has no attribute 'state_dict'