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

Question

使用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减小就不报错了。
但是是否会对模型训练有影响还未知。

Answer 1

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