The code in jupityer notebook S13Lightning.ipynb trains the ResNet model written in pytorch lightning.
Model: S13ResNetLightning_model
!pip install torch-lr-finder
!pip install grad-cam
!pip install pytorch-lightning
!git clone https://github.com/walnashgit/ERAV2_main.git
Follow the code in S13Lightning.ipynb for detailed code.
