I am trying to train a YoloV3 on a custom dataset by referencing  and modifying . In the model zoo, I saw that YoloV3 has 3 input dimensions: 310,416 and 608.
I tried training with input dimension 608 first, but after several epochs all losses were starting to report Nan, hence i switched to 416 and the losses were no longer reporting NaN. To note, I am using SGD with multi-precision: true
I am still looking to use 608 as my input size and was wondering if anyone would be able to offer guidance on the cause of the NaN issue