Fix: Requirements issue
Browse files- requirements.txt +6 -1
requirements.txt
CHANGED
@@ -4,4 +4,9 @@ torch-lr-finder
|
|
4 |
grad-cam
|
5 |
pillow
|
6 |
numpy
|
7 |
-
matplotlib
|
|
|
|
|
|
|
|
|
|
|
|
4 |
grad-cam
|
5 |
pillow
|
6 |
numpy
|
7 |
+
matplotlib
|
8 |
+
albumentations
|
9 |
+
seaborn
|
10 |
+
pandas
|
11 |
+
sklearn
|
12 |
+
torchmetrics
|