-
Notifications
You must be signed in to change notification settings - Fork 8k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Best weight check between training sessions and exported set_batch_network #3465
base: master
Are you sure you want to change the base?
Conversation
@keko950 could you please tell me, is the best file after training then always named |
Nope, final and best weights are different. |
Ok, I tested your PR and trained a model for 1000 batches. When the training was done, I got these 3 files: |
Could it be, that you first save a |
A |
So I tried again, I have now the following files:
Still no |
Just tested it again, working fine, trained 2000 iterations: What command are you running? Show me your ./darknet argument line |
darknet detector train darknet.data dnn.cfg weights/darknet53.conv.74
What do you mean? Of course I am running on your PR, since I want to test it |
Add -map option in order to generate best weights |
got it! Thanks! |
@AlexeyAB could you merge this please? |
ping |
@AlexeyAB would appreciate this too. Useful feature! |
Just modified detector.c in order to check the best weight map every time a training session is started as described in #3452.
If *best.weights exist in backup path, detector will do a map before start training, it will save it and initialize the best_map value, so now you can maintain your best weight across different training sessions :)
Edit:
Also updated network.h in order to add set_batch_network to the exported api, so you can use dynamic batching..