README.md

S13 Resent Lightning

The code in jupityer notebook S13Lightning.ipynb trains the ResNet model written in pytorch lightning.

Model: S13ResNetLightning_model

Demo: https://huggingface.co/spaces/walnash/erav2s13demo

Dependencies to install:

!pip install torch-lr-finder
!pip install grad-cam     
!pip install pytorch-lightning

Clone the main repo for model and training code:

!git clone https://github.com/walnashgit/ERAV2_main.git

Follow the code in S13Lightning.ipynb for detailed code.

LR Finder graph:

Training and vaidation accuracy at last epoch

Training Accuracy

Training Loss

Validation Accuracy

Validation loss

Miss-classified images

Miss-classified images with grad cam at -1 layer