Skip to content

Commit 9c01c4e

Browse files
authoredMay 11, 2024
Update README.md
1 parent 7a23697 commit 9c01c4e

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed
 

‎README.md

+15-1
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,25 @@
11
# S13 Resent Lightning
22

3-
## The code in jupityer notebook (S13Lightning.ipynb) trains a ResNet model written in pytorch lightning.
3+
## The code in jupityer notebook [S13Lightning.ipynb](https://github.com/walnashgit/S13Lightning/blob/main/S13Lightning.ipynb) trains the ResNet model written in pytorch lightning.
44

55
## Model: [S13ResNetLightning_model](https://github.com/walnashgit/ERAV2_main/blob/main/model/S13ResNetLightning_model.py)
66

77
## Demo: https://huggingface.co/spaces/walnash/erav2s13demo
88

9+
### Dependencies to install:
10+
11+
!pip install torch-lr-finder
12+
!pip install grad-cam
13+
!pip install pytorch-lightning
14+
15+
### Clone the main repo for model and training code:
16+
!git clone https://github.com/walnashgit/ERAV2_main.git
17+
18+
Follow the code in S13Lightning.ipynb for detailed code.
19+
20+
### LR Finder graph:
21+
<img width="582" alt="image" src="https://github.com/walnashgit/S13Lightning/assets/73463300/989eeea0-6088-4e5f-881c-ea066ed14574">
22+
923
### Training and vaidation accuracy at last epoch
1024

1125
<img width="1407" alt="image" src="https://github.com/walnashgit/S13Lightning/assets/73463300/ab9d8db3-7704-4439-95ea-12db73551fe3">

0 commit comments

Comments
 (0)
Please sign in to comment.