Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -3,6 +3,7 @@ torch>=2.0.0
|
|
| 3 |
torchvision>=0.15.0
|
| 4 |
lightning>=2.0.0
|
| 5 |
torchmetrics>=0.11.4
|
|
|
|
| 6 |
|
| 7 |
# --------- hydra --------- #
|
| 8 |
hydra-core==1.3.2
|
|
|
|
| 3 |
torchvision>=0.15.0
|
| 4 |
lightning>=2.0.0
|
| 5 |
torchmetrics>=0.11.4
|
| 6 |
+
wheel
|
| 7 |
|
| 8 |
# --------- hydra --------- #
|
| 9 |
hydra-core==1.3.2
|