Thomas.Chaigneau
commited on
Commit
•
9a5a46c
1
Parent(s):
60b60e4
fix file name and pin torch version
Browse files
requirement.txt → requirements.txt
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
-
torch
|
| 2 |
-
torchvision
|
| 3 |
pytorch-lightning
|
|
|
|
| 1 |
+
torch==1.10.1
|
| 2 |
+
torchvision==0.11.2
|
| 3 |
pytorch-lightning
|