Training time is larger than 15 minutes

deep-learning-with-pytorch / dlwpt-code

Code for the book Deep Learning with PyTorch by Eli Stevens, Luca Antiga, and Thomas Viehmann.

https://www.manning.com/books/deep-learning-with-pytorch

4.69k stars 1.98k forks source link

Training time is larger than 15 minutes #73

Open duydo77 opened 3 years ago

duydo77 commented 3 years ago

In section 11.7 (page 305), you said that "A single NVIDIA GTX 1080 Ti should complete an epoch in under 15 minutes". After running LunaPrepCacheApp(), it still predict training will finish after 2 days. I am training model with colab pro, which has Tesla P100. I want to know if I make a mistake or it's normal. I'm sorry for my dumy question.

duydo77 commented 3 years ago

I try to train with epochs=1, num_workers=4, batch_size=32