UserWarning: nn.init.kaiming_normal is now deprecated in favor of nn.init.kaiming_normal_.

基于torch1.8.1 训练模型时,出现如下warning,记录解决方法。

UserWarning: nn.init.kaiming_normal is now deprecated in favor of nn.init.kaiming_normal_.

解决:
出现警告的原因,是torch版本的更新导致之前的参数不再使用。然而警告并不影响代码运行,所以可以不管。但是,为了输出不再出现警告,可加入忽视warning的代码:

import warnings
warnings.filterwarnings('ignore')

你可能感兴趣的:(Pytorch,python,深度学习,深度学习,pytorch,神经网络)