Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -3,4 +3,5 @@ torchvision
|
|
3 |
grad-cam
|
4 |
pillow
|
5 |
numpy
|
6 |
-
albumentations
|
|
|
|
3 |
grad-cam
|
4 |
pillow
|
5 |
numpy
|
6 |
+
albumentations
|
7 |
+
pytorch-lightning
|