madhurjindal commited on
Commit
33c6562
1 Parent(s): 37887aa

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -1
requirements.txt CHANGED
@@ -5,4 +5,7 @@ matplotlib
5
  seaborn
6
  torchmetrics
7
  torch_lr_finder
8
- pytorch_lightning
 
 
 
 
5
  seaborn
6
  torchmetrics
7
  torch_lr_finder
8
+ pytorch_lightning
9
+ torchvision
10
+ albumentations
11
+ pytorch_grad_cam