yangwu commited on
Commit
094e7dd
1 Parent(s): dbed109

update requirements.txt

Browse files
Files changed (1) hide show
  1. 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
- segmentation_models_pytorch
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