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.
<div align="center">
<img alt="Lightning" src="https://pl-public-data.s3.amazonaws.com/assets_lightning/LightningColor.png" width="800px" style="max-width: 100%;">
<br/>
<br/>
**The deep learning framework to pretrain, finetune and deploy AI models.**
**NEW- Lightning 2.0 features a clean and stable API!!**
______________________________________________________________________
<p align="center">
<a href="https://lightning.ai/">Lightning.ai</a> •
<a href="https://lightning.ai/docs/pytorch/stable/">PyTorch Lightning</a> •
<a href="https://lightning.ai/docs/fabric/stable/">Fabric</a> •
<a href="https://lightning.ai/docs/app/stable/">Lightning Apps</a> •
<a href="https://pytorch-lightning.readthedocs.io/en/stable/">Docs</a> •
<a href="#community">Community</a> •
This file has been truncated. show original
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