TrainYourOwnYOLO icon indicating copy to clipboard operation
TrainYourOwnYOLO copied to clipboard

validation set and loss in tensorboard graph

Open joaoalves10 opened this issue 2 years ago • 4 comments

Hi @AntonMu i would like to ask you where i can find the graph related to the validation set? When i use the tensorboard only appears the loss of the trainning.

Another question is related to the loss graph from tensorboard. Can you explain to my why the loss has a such a steep decline between the first trainning epochs and the second ones?

I have another question, why do you freeze some layers in the first set of trainninf and then unfreeze them all in the last part? And how many layers do you freeze initially?

image

joaoalves10 avatar Apr 20 '22 09:04 joaoalves10

Everything you are asking, you can see in the colab: https://github.com/AntonMu/TrainYourOwnYOLO#google-colab-tutorial-

Hope this helps!

AntonMu avatar Apr 24 '22 00:04 AntonMu

it helps a bit. however i do not find the answer to why the loss has a such a steep decline between the first trainning epochs and the second ones? And why do you freeze some layers in the first set of trainninf and then unfreeze them all in the last part? And how many layers do you freeze initially?

joaoalves10 avatar Apr 24 '22 08:04 joaoalves10

If you look at the log messages under !python Train_YOLO.py, you see the exact numbers of frozen layers in each step.

Of course, if you freeze layers, you train only a small part of the model, and thus the loss is larger. When you unfreeze all layers, the model has more parameters to learn, and thus the loss is smaller.

AntonMu avatar Apr 25 '22 04:04 AntonMu

Thank you so much!

Em seg., 25 de abr. de 2022 às 05:07, AntonMu @.***> escreveu:

If you look at the log messages under !python Train_YOLO.py, you see the exact numbers of frozen layers in each step.

Of course, if you freeze layers, you train only a small part of the model, and thus the loss is larger. When you unfreeze all layers, the model has more parameters to learn, and thus the loss is smaller.

— Reply to this email directly, view it on GitHub https://github.com/AntonMu/TrainYourOwnYOLO/issues/250#issuecomment-1108053324, or unsubscribe https://github.com/notifications/unsubscribe-auth/AVMLLBIZ33LICRWU7IZOU6TVGYLAZANCNFSM5T3PH7TQ . You are receiving this because you authored the thread.Message ID: @.***>

joaoalves10 avatar Apr 25 '22 08:04 joaoalves10