RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be the same.

NZWANG / YOLOV7-AC

Object Detection and YOLOV7-AC

GNU General Public License v3.0

25 stars 4 forks source link

Open faiaz2011 opened 1 year ago

faiaz2011 commented 1 year ago

When I try to train using a custom dataset, I get the following error.

RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be the same.

Does anybody face a similar issue? any solution regarding this?

oneFU commented 1 year ago

我也是，您解决了吗？

faiaz2011 commented 1 year ago

我也是，您解决了吗？

No.