在进行去雨去雪去雾算法过程中,遇到了一个问题,这在先前的电脑运行是都没有出现过,但在博主新买的电脑上却出现了,讲道理是有点小抑郁的。
RuntimeWarning: invalid value encountered in scalar divide
ret = ret.dtype.type(ret / rcount)
NaN or Inf found in input tensor.
其实问题也很简单,就是由于除零导致的,但这种情况在博主看来是不太可能的,直到博主看到上面还有一个警告,这句话是一般找不到文件的警告提示。
RuntimeWarning: Mean of empty slice.
return _methods._mean(a, axis=axis, dtype=dtype,
然后再结合评估时的运行情况:
博主看了下验证数据集的设置情况,原来是文件名不一致。
修改完后便正常了。