search

Chris-hughes10 / Yolov7-training

A clean, modular implementation of the Yolov7 model family, which uses the official pretrained weights, with utilities for training the model on custom (non-COCO) tasks.
GNU General Public License v3.0
116 stars 35 forks source link

fix: :bug: fix device error in indexed tensor #5

Closed dariush-bahrami closed 1 year ago

dariush-bahrami commented 1 year ago

Regarding the issue in the original repo, I fixed it by ensuring the index tensor and the indexed tensor are both on the same device. This resolved the error for me.

Chris-hughes10 commented 1 year ago

Great, thanks for this!