tensorflow-yolov3 icon indicating copy to clipboard operation
tensorflow-yolov3 copied to clipboard

Training Loss: NAN (after some training) Test Loss: NAN (always)

Open MuhammadAsadJaved opened this issue 4 years ago • 3 comments

MuhammadAsadJaved avatar Aug 07 '20 03:08 MuhammadAsadJaved

Hi, I am training YOLOV3 with 1 class and dataset is about 7000 images.

The training loss decrease for few iterations and then stop decreasing, test loss is always NAN.
I am using default settings and anchors here are my configurations and loss reports. What is the possible reasons and how to overcome this problem? config loss

MuhammadAsadJaved avatar Aug 07 '20 03:08 MuhammadAsadJaved

change np.mean to np.nanmean like this: train_epoch_loss, test_epoch_loss = np.nanmean(train_epoch_loss), np.nanmean(test_epoch_loss)

MrZhousf avatar Sep 25 '20 07:09 MrZhousf

@MrZhousf OK. Thank you so much. I'll give it a try.

MuhammadAsadJaved avatar Sep 25 '20 10:09 MuhammadAsadJaved