Traceback (most recent call last):
File "main.py", line 98, in
train(cfg_param)
File "main.py", line 75, in train
trainer.run()
File "C:\Users\HP\Desktop\ws\python\YOLOV3\train\trainer.py", line 48, in run
self.run_iter()
File "C:\Users\HP\Desktop\ws\python\YOLOV3\train\trainer.py", line 37, in run_iter
self.yololoss.compute_loss(output, targets, self.model.yolo_layers)
File "C:\Users\HP\Desktop\ws\python\YOLOV3\train\loss.py", line 26, in compute_loss
tcls, tbox, tindices, tanchors = self.get_targets(pred, targets, yololayer)
File "C:\Users\HP\Desktop\ws\python\YOLOV3\train\loss.py", line 97, in gettargets
indices.append((b, a, gj.clamp(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1)))
RuntimeError: result type Float can't be cast to the desired output type __int64
Traceback (most recent call last): File "main.py", line 98, in
train(cfg_param)
File "main.py", line 75, in train
trainer.run()
File "C:\Users\HP\Desktop\ws\python\YOLOV3\train\trainer.py", line 48, in run
self.run_iter()
File "C:\Users\HP\Desktop\ws\python\YOLOV3\train\trainer.py", line 37, in run_iter
self.yololoss.compute_loss(output, targets, self.model.yolo_layers)
File "C:\Users\HP\Desktop\ws\python\YOLOV3\train\loss.py", line 26, in compute_loss
tcls, tbox, tindices, tanchors = self.get_targets(pred, targets, yololayer)
File "C:\Users\HP\Desktop\ws\python\YOLOV3\train\loss.py", line 97, in gettargets
indices.append((b, a, gj.clamp(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1)))
RuntimeError: result type Float can't be cast to the desired output type __int64