小编Abd*_*shr的帖子

我正在使用此代码在我的数据上训练 yolov3：https : //github.com/cfotache/pytorch_custom_yolo_training/

但我收到了这个烦人的弃用警告

Warning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead. (expandTensors at /pytorch/aten/src/ATen/native/IndexingUtils.h:20)

我尝试使用python3 -W ignore train.py 我尝试添加：

import warnings
warnings.filterwarnings('ignore')

但警告仍然存在。

我在 stackoverflow 上找到了这段代码，它在警告上打印了堆栈，

import traceback
import warnings
import sys

def warn_with_traceback(message, category, filename, lineno, file=None, line=None):

    log = file if hasattr(file,'write') else sys.stderr
    traceback.print_stack(file=log)
    log.write(warnings.formatwarning(message, category, filename, lineno, line))

warnings.showwarning = warn_with_traceback

这就是我得到的：

  File "/content/pytorch_custom_yolo_training/train.py", line 102, in <module>
   loss = model(imgs, targets)
  File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 532, …