Pytorch lightning slow on colab

I tried using pytorch lightning using the available tutirials. But the execution is very slow and after some time colab becomes unresponsive. Can anyone help me with some solution regarding this.

1 Like

Do you have an executable notebook on Colab?
Also, are you seeing this slow execution only with Lightning or also with plain PyTorch code?

I just used the official tutorial on MNIST classification.

It is the one in demo

GPU is enabled

Have you compared the code to a plain PyTorch implementation or how did you verify that it’s “slow”?

I have been training resnet20 on cifar10 in the same note book where I ran that particular code.
It was running fine using normal pytorch.

Thanks for the information.
CC @williamFalcon