pytorch(三)——保存自己的模型

本文详细介绍如何在PyTorch中保存模型参数及整个模型,包括两种保存方法的具体操作步骤,以及可能出现的错误及其解决办法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

pytorch(三)——保存自己的模型

模型依照前一片文章訓練好之後應該怎麼保存呢?

有兩個方法

1.保存模型參數


torch.save(the_model.state_dict(), PATH)


the_model = TheModelClass(*args, **kwargs)
the_model.load_state_dict(torch.load(PATH))

2.保存整個模型(推薦)


torch.save(the_model, PATH)

the_model = torch.load(PATH)

可能出現報錯

AttributeError: Can't get attribute 'Net' on module '__main__'
解決方法:補充上網絡的結構就可以了

測試

model = the_model()
print(model)

如果能打印出網絡結構,則成功了~~

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值