Pytorch中保存模型的两种方法-一、保存整个模型

这种方法会保存模型的整个定义(包括网络结构和参数)。这样保存的模型可以直接被加载并使用,但前提是保存和加载模型的代码环境需要完全一致。

保存模型时使用代码:

torch.save(model, 'model.pth')

加载模型时使用代码:

model = torch.load('model.pth')
上一篇:Vue从入门到精通-01-Vue的介绍和vue-cli


下一篇:Linux PTP学习