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

用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条回答

      报告相同问题?

      相关推荐 更多相似问题

      问题事件

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

      悬赏问题

      • ¥15 内存管理的一段代码不是很理解
      • ¥20 打开anaconda时卡在Loading applications无法进入界面
      • ¥15 网页超时时间设置失效
      • ¥15 有关绿色信贷毕业论文的问题
      • ¥30 关于#机器人#的问题,如何解决?
      • ¥15 求MATLAB函数ScalarLayerDisplay的代码
      • ¥15 安卓如何自动执行检测到的NFC标签,无需再点确认
      • ¥15 pyHM库mouse模块的ValueError错误
      • ¥15 python opencv 摄像头 传感器
      • ¥30 eMMC&Android&C&Linux