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
|
||||
tensorflow==1.6.0
|
||||
threadpool==1.3.2
|
||||
torch==1.2.0
|
||||
torchvision==0.4.0
|
||||
|
|
Loading…
Reference in New Issue