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