保存模型：

torch.save({'epoch': epoch + 1,'state_dict': model.state_dict(),'optimizer': optimizer.state_dict(),}, datadir)

加載模型

model = model_class(num_classes=num_classes) # 定義模型 state = torch.load(datadir) model.load_state_dict(state['state_dict'])

總結(jié)

以上是生活随笔為你收集整理的pytorch保存和加载模型state_dict的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。