去掉执行python代码终端warning警告提示的方法

问题描述

在终端执行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")

不过警告还是需要引起重视的，短暂的忽略掉，后面也许有其他问题不好排查~