Training issues

[daiisuke27]

Hello, i am running your code on Colab.

• My issue is that i am getting nan values in the training process.

• I want also to know how do you define the logs keys, because i want to show the obj_loss and class_loss values in callbacks, but i cant find a way to do it.

Thank you.

[RobotEdh]

hello,

1/ which training set do you use ? I never get nan

2/ my loss fucntion returns only xy_loss + wh_loss +obj_loss +class_loss , so you can't get only obj_loss or class_loss, why do you need them ?

[RobotEdh]

in fact, regrading the nan issue, i have the same error when i activate GPU. but in in this case i have also error to allocate memory for convolution redzone , maybe it is the root cause of the nan

[RobotEdh]

I have comitted a fix, you can try again