yolov7训练出错

[jo-dean]

命令：python train.py --mode yolov7 --use_aux True --data garbage.yaml --cfg yolov7.yaml --weights weights/yolov7/yolov7.pt --batch-size 8
出现错误
**Logging results to runs/train/exp14
Starting training for 100 epochs...

 Epoch   gpu_mem       box       obj       cls    labels  img_size

0%| | 0/5 [00:01<?, ?it/s]
Traceback (most recent call last):
File "train.py", line 772, in
main(opt)
File "train.py", line 670, in main
train(opt.hyp, opt, device, callbacks)
File "train.py", line 448, in train
loss, loss_items = compute_loss(pred, targets.to(device), imgs) # init loss class
File "YOLOU/utils/loss.py", line 992, in call
bs_aux, as_aux_, gjs_aux, gis_aux, targets_aux, anchors_aux = self.build_targets2(p[:self.nl], targets, imgs)
File "YOLOU/utils/loss.py", line 1202, in build_targets2
indices, anch = self.find_5_positive(p, targets)
File "YOLOU/utils/loss.py", line 1388, in find_5_positive
indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1))) # image, anchor, grid indices
RuntimeError: result type Float can't be cast to the desired output type long int
**