Pytorch 训练时如何屏蔽warning

你是否为训练时，弹出很多warning而烦恼？一个epoch拉满一页，怎么做呢？

代码输出过滤

import warning

warning.filterwarning('ignore')

命令行输出过滤

执行python文件时添加参数过滤

python -W ignore your_script.py

参考

Python3标准库 - 使用模式过滤