Skip to content
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

Continue training error and trained model #19

Open
Star1ess opened this issue Apr 22, 2023 · 2 comments
Open

Continue training error and trained model #19

Star1ess opened this issue Apr 22, 2023 · 2 comments

Comments

@Star1ess
Copy link

In train.py line 137
for epoch in range(start_epoch, args.epochs):
When I want to continue to train based on a model weight, there will be an error. Maybe it should be start_epoch+args.epochs ?

I also woud like to ask if you can provide a trained weight, which can be very convenient.

Thank you

@gwinndr
Copy link
Owner

gwinndr commented Apr 22, 2023

what is the error? Also yeah I should probably re-train the model... the weights I got are unfortunately lost.

@Star1ess
Copy link
Author

For example, I first trained 20 epochs and the next time I want to continue to train 20 more epochs based on the last weight "epoch_0020.pickle". So the argument should be -continue_weights ...\epoch_0020.pickle -continue_epoch 20 -epochs 20. So for for epoch in range(start_epoch, args.epochs): the range will be (20, 20), and the script will not run. I do not know if my understanding of -continue_epoch arugment function is correct. Maybe it is my mistake.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants