positive666/yolo_research

RuntimeError: expected scalar type Float but found Half

shenshi19 opened this issue · 0 comments

Epoch gpu_mem box obj cls labels img_size
0/199 4.87G 0.1106 0.01998 0 13 640: 100%|██████████| 567/567 [01:50<00:00, 5.14it/s]
Class Images Labels P R mAP@.5 mAP@.5:.95: 0%| | 0/78 [00:00<?, ?
Traceback (most recent call last):
File "train.py", line 675, in
main(opt)
File "train.py", line 572, in main
train(opt.hyp, opt, device, callbacks)
File "train.py", line 396, in train
results, maps, _ = val.run(data_dict,
RuntimeError: expected scalar type Float but found Half
大佬,所有的swin都会出现这个问题是什么原因呢