-
Notifications
You must be signed in to change notification settings - Fork 67
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
How to choose the best model? #5
Comments
Sorry, we only provided a simple version of the code. You can add tensorboard to your code according to your needs. In fact, we usually choose the latest model for testing. |
I have trained 30 epochs. I remember the result in your paper is gotten in 10 epochs on OpenLane dataset. |
We found that the effects of the model of 10th epoch and models of continue training were not significantly different. |
And I find it seems that your codes can't resume training though there is a funtion called "resume_training". |
@ppbangKGT it seems so |
There are some small changes to do to use it: def save_model_dp torch.save({ def load_model(model, model_state_file): |
Yes, there can be different methods to resume training. What I do is :
|
@ppbangKGT Can you share the source code? Thank you very much,1017094591@qq.com |
@ppbangKGT hello, would you please share me the source code? 405612048@qq.com |
@ppbangKGT hello, would you please share me the source code? 1653658300@qq.com |
@ppbangKGT hello, would you please share me the source code? hitbuyi@163.com |
Hello, I am a new researcher of 3D-BEV-LaneDet, could you please share the source code with me? |
Thanks for your open-source codes. When I train the model, I find that it seems that you don't use tensorboard to record the loss or F1-score. And the total performance of one epoch isn't recorded either. So I wonder how to choose the best model I have got. Should I run the val_openlane.py on each of the models I trained? Or maybe I miss something important.
Thanks for your reply.
The text was updated successfully, but these errors were encountered: