UserWarning: nn.init.xavier_uniform is now deprecated in favor of nn.init.xavier_uniform_

文章目录

  • warning背景:
  • UserWarning:
  • 解决方案:

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_ .

解决方案:

  1. 首先代码运行过程出现warning是不会影响代码运行的,所以可以完全不管。
  2. 忽视warning,加入下面两句代码即可让程序不报warning。
    import warnings
    warnings.filterwarnings(‘ignore’)
  3. 解决:其实warning中讲的很清楚,“nn.init.xavier_uniform”不被推荐使用,建议使用“nn.init.xavier_uniform_”,在代码中改掉即可,另外两个同理。
    究其出现的原因,就是“nn.init.xavier_uniform”是torch以前的版本使用的,在torch1.7.0版本中
    已经被弃用。

你可能感兴趣的:(工具类,深度学习,pytorch)