Model takes too much time to train

[SailSabnis]

Hi all,

I wrote the exact same code as Daniel and trying to run it on 2 epochs. But, it takes forever. I am on CPU but still it takes unusually long time. Is this normal?

[LuluW8071]

It depends on hardware, data and model complexity u r training on. CPU generally takes longer time if there are lots of featured data and huge parameter in model. Try increasing batch size and decrease learning rate if it still persists OR Restart the kernel to refresh RAM and run again.

[SailSabnis]

Many thanks! will give it a try

[mrdbourke]

Hi @SailSabnis ,

If you are using a CPU for larger datasets, your model will likely take quite a bit longer to train.

GPU is best for larger datasets/models.

I'd only recommend CPU for smaller deep learning models/datasets.

[SailSabnis]

thinking of getting Colab pro (for a few months atleast) to avoid lot of time waste. Even 5 epochs take >25mins on CPU. It is frustrating.