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