Expected all tensors to be on the same device ,but found at least two devices,cuda:0 and cpu - Githubissues

Javacr / PyQt5-YOLOv5

PyQt5 implementation of YOLOv5 GUI

GNU General Public License v3.0

1.32k stars 222 forks source link

Expected all tensors to be on the same device ,but found at least two devices,cuda:0 and cpu #29

Open ruanzhijian opened 1 year ago