在使用pycharm+pytorch进行YOLOV3-spp进行预训练权重进行训练时出现报错

Question

在使用pycharm+pytorch进行YOLOV3-spp进行预训练权重进行训练时出现报错
报错内容如下：
Traceback (most recent call last):
File "E:\PycharmProjects\deep-learning-for-image-processing-master\pytorch_object_detection\yolov3_spp\train.py", line 302, in
train(hyp)
File "E:\PycharmProjects\deep-learning-for-image-processing-master\pytorch_object_detection\yolov3_spp\train.py", line 208, in train
scaler=scaler)
File "E:\PycharmProjects\deep-learning-for-image-processing-master\pytorch_object_detection\yolov3_spp\train_utils\train_eval_utils.py", line 61, in train_one_epoch
loss_dict = compute_loss(pred, targets, model)
File "E:\PycharmProjects\deep-learning-for-image-processing-master\pytorch_object_detection\yolov3_spp\build_utils\utils.py", line 213, in compute_loss
tcls, tbox, indices, anchors = build_targets(p, targets, model) # targets
File "E:\PycharmProjects\deep-learning-for-image-processing-master\pytorch_object_detection\yolov3_spp\build_utils\utils.py", line 297, in build_targets
a, t = at[j], t.repeat(na, 1, 1)[j] # filter
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

Answer 1

不知道你这个问题是否已经解决, 如果还没有解决的话:

• 这篇文章：pycharm搭建yolo3-pytorch-master的pytorch环境 也许有你想要的答案，你可以看看

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^