Unstable training

[harrylin-hyl]

I set the number of GPUs to 1, and then the predicted quality values of iou all become 0, making the training unable to proceed normally. Whether it can be improved without increasing the GPUs Thanks!