神经网络参数初始化及训练过程loss出现NAN的解决方法
一、最近在修改代码后,训练过程loss出现NAN,而且概率很大,于是查找原因,在此记录。在给信号按照相应信噪比加噪声的过程中,增加了部分代码,且增加了batch norm,但是训练过程中大概率loss会出现NAN。增加噪声的代码如下:1 def wgn(x, snr):2 P_signal = np.sum(abs(x)**2)/len(x)3 P_noise = P_signal/10**(snr/10.0)4 return np....
原创
2020-10-14 15:29:18 ·
1719 阅读 ·
1 评论