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

Training problem #21

Open
HasilPark opened this issue Aug 24, 2021 · 1 comment
Open

Training problem #21

HasilPark opened this issue Aug 24, 2021 · 1 comment

Comments

@HasilPark
Copy link

I trained by setting the epoch parameter to 20 using got-10k dataset.
However, only epochs 1 to 19 are saved.
What's the problem?

@HasilPark HasilPark reopened this Aug 24, 2021
@Capricorn231
Copy link
Collaborator

When the total training data can be divisible by batch_size, the model of the 20th epoch cannot be saved. This is a small problem in our code, which will be optimized later. You can save the model of the 20th epoch by adjusting the batch_size.

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