对于输出nan

在python中,nan不会报错,只会累加,最后才会输出nan

解决办法:
可以先判断再去掉

if torch.isnan(cam).sum()==0:

你可能感兴趣的:(对于输出nan)