warning背景:
基于torch1.7.0 训练模型时,出现如下warning,记录解决方法。
UserWarning:
出现如下的UserWarning:
1)UserWarning: nn.init.xavier_uniform is now deprecated in favor of nn.init.xavier_uniform_ .
2) UserWarning: nn.init.orthogonal is now deprecated in favor of nn.init.orthogonal_ .
3) UserWarning: nn.init.constant is now deprecated in favor of nn.init.constant_ .
解决方案:
- 首先代码运行过程出现warning是不会影响代码运行的,所以可以完全不管。
- 忽视warning,加入下面两句代码即可让程序不报warning。
import warnings
warnings.filterwarnings(‘ignore’) - 解决:其实warning中讲的很清楚,“nn.init.xavier_uniform”不被推荐使用,建议使用“nn.init.xavier_uniform_”,在代码中改掉即可,另外两个同理。
究其出现的原因,就是“nn.init.xavier_uniform”是torch以前的版本使用的,在torch1.7.0版本中
已经被弃用。