问题描述
在终端执行Python代码时候,终端会输出warning警告。如:
UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument.
output = self.softmax(output)
原因分析:
出现这个警告的原因主要是Pytorch中的Softmax()函数要加上有指明dim参数。这个写法在早期的Pytorch版本是没有警告的,现在虽然程序仍可以成功运行成功,但是会有warning警告提示。
可将 nn.Softmax()
改成 nn.Softmax(dim=1)
这不是本篇的重点,本篇的重点是不希望在终端出现警告提示。
解决方案:
如果希望终端不输出警告,可以在终端执行代码时或在程序中去掉警告。
在终端执行Python代码可以加上 -W ignore
参数:
python -W ignore main.py
在程序中去掉警告的方式是:
import warnings
warnings.filterwarnings("ignore")
不过警告还是需要引起重视的,短暂的忽略掉,后面也许有其他问题不好排查~