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.

Sign up for GitHub

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

Jump to bottom

Continue training? #8

Open
hiepph opened this issue Feb 12, 2018 · 2 comments
Open

Continue training? #8

hiepph opened this issue Feb 12, 2018 · 2 comments
Labels
enhancement help wanted

Comments

@hiepph
Copy link

hiepph commented Feb 12, 2018

Train CapsulesNet is known as time-consuming and not yet optimized. So, I want an option of loading pre-trained weights (e.g. provided file path) and continue training.
@cedrickchee
Copy link
Owner

What is the training speed looks like for you? I mean, how many minutes per epoch and what GPU are you using? I will try to optimize the code soon. The option to load pre-trained weights and continue training is on my todo for some time. Sorry about that. I will try to work on this next week but no promise.

@hiepph
Copy link
Author

hiepph commented Apr 14, 2018

My training speed so far

GPU Min/epoch
Geforce GTX 960 11
Geforce GTX 1080Ti 2

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement help wanted
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@cedrickchee @hiepph