leobcc
commited on
Commit
·
35ab1e2
1
Parent(s):
798f9a0
Updated requirement.txt
Browse files- requirement.txt +2 -1
requirement.txt
CHANGED
|
@@ -1,7 +1,8 @@
|
|
| 1 |
cython
|
| 2 |
chumpy==0.70
|
| 3 |
scikit-image==0.18.1
|
| 4 |
-
torch==1.9.1
|
|
|
|
| 5 |
pytorch-lightning==1.5.7
|
| 6 |
hydra-core==1.1.1
|
| 7 |
wandb==0.13.1
|
|
|
|
| 1 |
cython
|
| 2 |
chumpy==0.70
|
| 3 |
scikit-image==0.18.1
|
| 4 |
+
torch==1.9.1
|
| 5 |
+
torchvision==0.10.1
|
| 6 |
pytorch-lightning==1.5.7
|
| 7 |
hydra-core==1.1.1
|
| 8 |
wandb==0.13.1
|