qq_42519463 2022-08-26 10:04 采纳率: 62.5%
浏览 346
已结题

用torchsummary.summary时,报错溢出:RuntimeWarning: overflow encountered in long_scalars

使用torchsummary.summary时,报错溢出:RuntimeWarning: overflow encountered in long_scalars
关键是如果不用torchsummary.summary,就不会报错,如下:
这样写是不报错的,如下

model3 = FeatureFuse(64)
y = model3(torch.rand(1,64, 100, 100),torch.rand(1,64, 100, 100))
print(y.size())

如果用torchsummary.summary,
summary(model3, [(64, 100, 100), (64, 100, 100)], batch_size=5, device="cpu")

就会报溢出错误,如下
RuntimeWarning: overflow encountered in long_scalars
total_input_size = abs(np.prod(input_size) * batch_size * 4. / (1024 ** 2.))
FeatureFuse模块里就是一些注意力机制模块,并没有涉及其他解决办法里提到的numpy之类的,这种情况的错误有解决办法吗?或者可以不解决吗?忽略这个问题会对结果有影响吗?

更新:是abs(np.prod(input_size) * batch_size * 4. / (1024 * 2.))这个数值太大,使用torchsummary.summary时把输入尺寸减小、或batch_size减小就不报错了。
但是是否会对模型训练有影响还未知。

  • 写回答

2条回答 默认 最新

  • 万里鹏程转瞬至 人工智能领域优质创作者 2022-08-29 08:16
    关注

    不影响,这里好像是数值溢出,就用int8去表示int16的数值一样。减小batch_size能跑,就表示没有数值溢出,那就是正常的。你也可用找到报错的哪行代码强制把int32转成int64.

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)
编辑
预览

报告相同问题?

问题事件

  • 系统已结题 9月13日
  • 已采纳回答 9月6日
  • 修改了问题 8月26日
  • 创建了问题 8月26日

悬赏问题

  • ¥15 centos7.6进不去系统,卡在数字7界面
  • ¥15 Tensorflow采用interpreter.allocate_tensors()分配内存出现ValueError: vector too long报错
  • ¥15 使用CGenFF在线生成血红素辅基拓扑结构遇到问题
  • ¥15 在fragment使用okhttp同步上传文件,能不能在fragment销毁后还可以继续上传文件?
  • ¥20 matlab代码实现可达矩阵形成骨骼矩阵
  • ¥15 关于地板的木纹和图库中的匹配的
  • ¥30 机器学习预测疾病模型流程疑问
  • ¥50 2048Python实现
  • ¥15 使用ads进行低噪放仿真没有结果且不报错
  • ¥15 关于#python#的问题:有偿求一个千寻框架找书机器人插件
手机看
程序员都在用的中文IT技术交流社区

程序员都在用的中文IT技术交流社区

专业的中文 IT 技术社区,与千万技术人共成长

专业的中文 IT 技术社区,与千万技术人共成长

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

客服 返回
顶部