bala1802 commited on
Commit
3a3f76e
1 Parent(s): 5a3d8ad

added requirement.txt and .gitignore files

Browse files
Files changed (2) hide show
  1. .gitignore +0 -0
  2. requirements.txt +11 -0
.gitignore ADDED
File without changes
requirements.txt ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ torch==2.0.1
2
+ torchvision==0.15.2
3
+ lightning==2.0.6
4
+ albumentations==1.3.1
5
+ torch-lr-finder
6
+ torchinfo
7
+ grad-cam
8
+ torchmetrics==0.8.0
9
+ pandas
10
+ datasets==2.14.4
11
+ tokenizers==0.13.3