在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刪除。