基于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')

在使用torch1.8.1进行模型训练时遇到UserWarning:nn.init.kaiming_normal已弃用,推荐使用nn.init.kaiming_normal_。警告不影响代码执行,但为消除警告,可以在代码中导入warnings模块并设置filterwarnings忽略警告。
1466

被折叠的 条评论
为什么被折叠?



