update reqs
Browse files- requirements.txt +11 -1
requirements.txt
CHANGED
@@ -1 +1,11 @@
|
|
1 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
torch
|
2 |
+
kornia
|
3 |
+
omegaconf
|
4 |
+
pytorch-lightning
|
5 |
+
torchvision
|
6 |
+
tqdm
|
7 |
+
torchmetrics
|
8 |
+
scikit-learn
|
9 |
+
numpy
|
10 |
+
matplotlib
|
11 |
+
timm==0.4.12
|