tensorflow-yolov4-tflite icon indicating copy to clipboard operation
tensorflow-yolov4-tflite copied to clipboard

nan Error while applying Transfer learning on pre-trained weights

Open dinesh-tadi opened this issue 3 years ago • 0 comments

Getting nan as outputs of the model if I load pre-trained weights (yolov4.weights or yolov4.conv.137).

Also getting nan in the training function (losses) after a few iterations if proceeding without loading pre-trained weights.

please suggest any fixes.

dinesh-tadi avatar Jun 06 '22 09:06 dinesh-tadi