东荷新绿 2022-04-18 11:28 采纳率: 0%
浏览 245

torch.save保存权重文件和net.parameters()参数量大小不一致

torch保存权重和参数量大小不一致
问题说明及相关代码

采用param.numel()计算参数

 for param in net.parameters():
      num_params += param.numel()

结果如下:

img

采用torch.save()保存权重大小

torch.save(net.cpu().state_dict(), save_path)

结果如下:

img

网上的部分解释

img

发现两者的大小关系确实接近四倍,但其中具体的细节不清楚,希望帮忙解答下疑惑,详细解释下,万分感谢。

  • 写回答

3条回答 默认 最新

  • 不会长胖的斜杠 后端领域新星创作者 2022-04-18 11:36
    关注

    这个是不是和文件占用空间和文件大小不一致同理呢

    评论

报告相同问题?

问题事件

  • 创建了问题 4月18日

悬赏问题

  • ¥15 随身WiFi网络灯亮但是没有网络,如何解决?
  • ¥15 gdf格式的脑电数据如何处理matlab
  • ¥20 重新写的代码替换了之后运行hbuliderx就这样了
  • ¥100 监控抖音用户作品更新可以微信公众号提醒
  • ¥15 UE5 如何可以不渲染HDRIBackdrop背景
  • ¥70 2048小游戏毕设项目
  • ¥20 mysql架构,按照姓名分表
  • ¥15 MATLAB实现区间[a,b]上的Gauss-Legendre积分
  • ¥15 delphi webbrowser组件网页下拉菜单自动选择问题
  • ¥15 linux驱动,linux应用,多线程