Problem with training Scaled-YOLOv4 Pytorch on custom dataset tutorial

[Mohamed]

@axlecky here is how to solve the issue, fully diagnosed:

Error: Roboflow-Custom-Scaled-YOLOv4

I found the Issue also reported in the YOLOv7 repo (creator of Scaled-YOLOv4 also made YOLOv7). It seems they introduced the same error in YOLOv7. This comment notes how to fix it: RuntimeError: result type Float can't be cast to the desired output type long int · Issue #35 · WongKinYiu/yolov7 · GitHub The fix is to convert to long the "gain" of the gridspace on declaration. In all lines with gain = torch.ones(7, device=targets.device) change to gain = torch.ones(7, device=targets.device).…

Error: Roboflow-Custom-Scaled-YOLOv4

Updated ScaledYOLOv4/utils/general.py (Google Drive download) Once this PR is merged to the yolov4-large branch of the Scaled-YOLOv4 repo fork, you’ll be able to run the notebook while avoiding making any of the manual changes noted above ^ : update line 512 and line 1101 of utils/general.py by mo-traor3-ai · Pull Request #2 · roboflow/ScaledYOLOv4 · GitHub