RobotEdh / Yolov-4

Yolo v4 using TensorFlow 2.x
110 stars 30 forks source link

Training issues (GPU Tensorflow) #19

Open adriacomos opened 2 years ago

adriacomos commented 2 years ago

Maybe it's been a while since this forum was updated, but I'll try ;).

I am trying to train the network. For this I have followed the YoloV4_Train_tf.ipynb source. I've had to change a few things: it seems that the mish function was missing to be registered, and I've tried it with and without, the latter changing the activation on the network to weakyrReLU). In both cases, I find very high losses that are difficult to lower (it starts at 50's and stays around 25's... and very low accuracy = 0.00...). In fact...the accuracy is decreasing! What could by happening?

thanks in advance