add torch in requirements.txt
This commit is contained in:
parent
c83f2649b6
commit
bf33be0e27
|
@ -19,3 +19,5 @@ scikit-learn==0.19.1
|
||||||
scipy==1.0.0
|
scipy==1.0.0
|
||||||
tensorflow==1.6.0
|
tensorflow==1.6.0
|
||||||
threadpool==1.3.2
|
threadpool==1.3.2
|
||||||
|
torch==1.2.0
|
||||||
|
torchvision==0.4.0
|
||||||
|
|
Loading…
Reference in New Issue