capsule-net-pytorch icon indicating copy to clipboard operation
capsule-net-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon cedrickchee

Continue training?

Open hiepph opened this issue 7 years ago • 2 comments

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.

hiepph avatar Feb 12 '18 08:02 hiepph

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.

cedrickchee avatar Apr 14 '18 03:04 cedrickchee

My training speed so far

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

hiepph avatar Apr 14 '18 04:04 hiepph