zhangshengdong commited on
Commit
5ef3c1a
·
1 Parent(s): 6f44353

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- torch
2
- torchvision
3
  # opencv
4
  opencv-contrib-python
5
  segmentation-models-pytorch==0.2.1
 
1
+ torch==1.10.2
2
+ torchvision==0.11.3
3
  # opencv
4
  opencv-contrib-python
5
  segmentation-models-pytorch==0.2.1