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

用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 16:16
    关注

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

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

报告相同问题?

问题事件

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

悬赏问题

  • ¥15 小程序中fit格式等运动数据文件怎样实现可视化?(包含心率信息))
  • ¥15 如何利用mmdetection3d中的get_flops.py文件计算fcos3d方法的flops?
  • ¥40 串口调试助手打开串口后,keil5的代码就停止了
  • ¥15 电脑最近经常蓝屏,求大家看看哪的问题
  • ¥60 高价有偿求java辅导。工程量较大,价格你定,联系确定辅导后将采纳你的答案。希望能给出完整详细代码,并能解释回答我关于代码的疑问疑问,代码要求如下,联系我会发文档
  • ¥50 C++五子棋AI程序编写
  • ¥30 求安卓设备利用一个typeC接口,同时实现向pc一边投屏一边上传数据的解决方案。
  • ¥15 SQL Server analysis services 服务安装失败
  • ¥15 基于面向对象的图书馆借阅管理系统
  • ¥15 opencv图像处理,需要四个处理结果图