deep-learning-for-image-processing icon indicating copy to clipboard operation
deep-learning-for-image-processing copied to clipboard
verified publisher icon WZMIAOMIAO

yolov3_spp正负样本匹配device设置

Open jiaerfei opened this issue 2 years ago • 2 comments

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)

jiaerfei avatar Mar 02 '23 01:03 jiaerfei

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

fengjingshuaige avatar Oct 13 '23 10:10 fengjingshuaige

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

祝您健康、快乐！

哈尔滨工业大学

learning-art avatar Oct 13 '23 10:10 learning-art