add torch
Browse files- requirements.txt +3 -0
requirements.txt
CHANGED
@@ -21,5 +21,8 @@ scipy==1.9.1
|
|
21 |
setuptools==59.5.0
|
22 |
timm==0.6.13
|
23 |
tqdm==4.65.0
|
|
|
|
|
|
|
24 |
detectron2@git+https://github.com/facebookresearch/detectron2.git@4e447553eb32b6e3784df0b8fca286935107b2fd
|
25 |
git+https://github.com/jinlinyi/PerspectiveFields.git@dev#egg=perspective2d
|
|
|
21 |
setuptools==59.5.0
|
22 |
timm==0.6.13
|
23 |
tqdm==4.65.0
|
24 |
+
|
25 |
+
torch==1.10.0
|
26 |
+
torchvision==0.11.1
|
27 |
detectron2@git+https://github.com/facebookresearch/detectron2.git@4e447553eb32b6e3784df0b8fca286935107b2fd
|
28 |
git+https://github.com/jinlinyi/PerspectiveFields.git@dev#egg=perspective2d
|