search

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite
https://github.com/hunglc007/tensorflow-yolov4-tflite
MIT License
2.24k stars 1.24k forks source link

nan Error while applying Transfer learning on pre-trained weights #447

Open D10752002 opened 2 years ago

D10752002 commented 2 years ago

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.