Cyril666 commited on
Commit
7624776
·
1 Parent(s): 8decb50

First model version

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -4,8 +4,8 @@ cython
4
  matplotlib
5
  tqdm
6
  opencv-python
7
- torch
8
- torchvision
9
  pytorch-extension
10
  shapely
11
  pyclipper
 
4
  matplotlib
5
  tqdm
6
  opencv-python
7
+ torch==1.6.0
8
+ torchvision==0.7.0
9
  pytorch-extension
10
  shapely
11
  pyclipper