Spaces:
Build error
Build error
yangwu
commited on
Commit
•
094e7dd
1
Parent(s):
dbed109
update requirements.txt
Browse files- requirements.txt +5 -5
requirements.txt
CHANGED
@@ -1,17 +1,17 @@
|
|
1 |
requests
|
2 |
-
torch
|
3 |
-
torchvision
|
4 |
-
pytorch-lightning
|
5 |
lightning-bolts
|
6 |
tqdm
|
7 |
wandb
|
8 |
-
|
9 |
albumentations==0.4.3
|
10 |
pandas
|
11 |
loguru
|
12 |
matplotlib
|
13 |
tensorboard
|
14 |
-
mmcv
|
15 |
docutils
|
16 |
PyYAML
|
17 |
prettytable
|
|
|
1 |
requests
|
2 |
+
torch==1.10.1+cu111
|
3 |
+
torchvision==0.11.2+cu111
|
4 |
+
pytorch-lightning==1.6.5
|
5 |
lightning-bolts
|
6 |
tqdm
|
7 |
wandb
|
8 |
+
segmentation-models-pytorch==0.2.1
|
9 |
albumentations==0.4.3
|
10 |
pandas
|
11 |
loguru
|
12 |
matplotlib
|
13 |
tensorboard
|
14 |
+
mmcv==1.6.1
|
15 |
docutils
|
16 |
PyYAML
|
17 |
prettytable
|