Skip to content

Commit

Permalink
add pytorch in requirements.txt
Browse files Browse the repository at this point in the history
  • Loading branch information
@taochenshh
taochenshh committed Jun 28, 2020
1 parent 62bba12 commit c425e22
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions requirements.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,4 +8,6 @@ cloudpickle
gitpython
tensorboard
tqdm
torch==1.4.0
torchvision==0.5.0
dataclasses;python_version < '3.7'

0 comments on commit c425e22

Please sign in to comment.