1条回答 默认 最新
关注 - 你看下这篇博客吧, 应该有用👉 :PyTorch模型保存torch.save()与加载torch.load()
- 你还可以看下pytorch参考手册中的 pytorch save_to_buffer() (torch.jit.ScriptFunction method)
- 除此之外, 这篇博客: PyTorch模型保存torch.save()与加载torch.load()中的 2)保存部分 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
只保存模型训练的参数(不包括网络结构)
torch.save(model.state_dict(), path)
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报
悬赏问题
- ¥15 linux驱动,linux应用,多线程
- ¥20 我要一个分身加定位两个功能的安卓app
- ¥15 基于FOC驱动器,如何实现卡丁车下坡无阻力的遛坡的效果
- ¥15 IAR程序莫名变量多重定义
- ¥15 (标签-UDP|关键词-client)
- ¥15 关于库卡officelite无法与虚拟机通讯的问题
- ¥15 目标检测项目无法读取视频
- ¥15 GEO datasets中基因芯片数据仅仅提供了normalized signal如何进行差异分析
- ¥100 求采集电商背景音乐的方法
- ¥15 数学建模竞赛求指导帮助