add torch in requirements.txt

This commit is contained in:
Dongsheng Yang 2019-09-22 17:23:53 -04:00 committed by Dana Van Aken
parent c83f2649b6
commit bf33be0e27
1 changed files with 2 additions and 0 deletions

View File

@ -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