search

WZMIAOMIAO / deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.
GNU General Public License v3.0
22.4k stars 7.91k forks source link

yolov3_spp正负样本匹配device设置 #721

Open jiaerfei opened 1 year ago

jiaerfei commented 1 year ago

https://github.com/WZMIAOMIAO/deep-learning-for-image-processing/blob/80681e81327320a37cfc35b754d8204a5156e9c2/pytorch_object_detection/yolov3_spp/build_utils/utils.py#L287 在跑模型时跳的bug，at的device为cpu，下面需要添加一行 at = at.to(targets.device)

fengjingshuaige commented 11 months ago

你好，请问改在哪里添加呢

learning-art commented 11 months ago

您给我发的信息已收到！ 我会及时回复信息！

祝您健康、快乐！

哈尔滨工业大学