点此进入CSDN

点此添加QQ好友 加载失败时会显示




2020年9月3日
pytorch两种模型保存方式
摘要： 只保存模型参数 # 保存 torch.save(model.state_dict(), '\parameter.pkl') # 加载 model = TheModelClass(...) model.load_state_dict(torch.load('\parameter.pkl')) 保存完整 阅读全文
posted @ 2020-09-03 11:02 高颜值的殺生丸 阅读(291) 评论(0) 推荐(0) 编辑

作者信息

昵称：

刘新宇

园龄：4年6个月


粉丝：1209


QQ：522414928