在pytorch中有两种模型保存方式，如下所示：
# 第一种：只保留模型参数：
torch.save(model.state_dict(),path) # 保存
model.load_state_dict(torch.load(path)) # 加载（需提前定义模型）
# 第二种：保存整个模型：
torch.save(model,path) # 保存模型
model = torch.load(path) # 加载模型
pytorch中保存模型
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。