探头的新芽

关注

[pytorch][原创]保存模型和权重

探头的新芽

关注

阅读 9

2024-11-04

只保存模型参数
 # 保存
 torch.save(model.state_dict(), '\model.pth')
 # 加载
 model = TheModelClass(...)
 model.load_state_dict(torch.load('\model.pth'))
 保存完整模型
 # 保存
 torch.save(model, '\model.pth')
 # 加载
 model = torch.load('\model.pth')

相关推荐

四月Ren间

pytorch-模型保存

四月Ren间 57 0 0

技术只适用于干活

PyTorch 06: PyTorch保存和加载模型

技术只适用于干活 200 0 0

颜娘娘的碎碎念

【Pytorch】保存和加载模型

颜娘娘的碎碎念 104 0 0

前端王祖蓝

Pytorch模型保存与加载

前端王祖蓝 85 0 0

河南妞

PyTorch保存训练好的模型

河南妞 31 0 0

color_小浣熊

【pytorch】(六)保存和加载模型

color_小浣熊 134 0 0

英乐

pytorch jit trace gpu 保存模型

英乐 15 0 0

惠特曼

学习pytorch17 pytorch模型保存及加载

惠特曼 42 0 0

奔跑的酆

pytorch学习001- -如何保存模型

奔跑的酆 58 0 0

把平平凡凡的生活过成诗与远方

PyTorch|保存及加载模型、nn.Sequential、ModuleList和ModuleDict

把平平凡凡的生活过成诗与远方 50 0 0

精彩评论(0)

0 0 举报