Add back torch/torchvision requirements

f8fa1325 · Jeff Rasley · GitHub · 6cb332f1 · f8fa1325
Unverified Commit f8fa1325 authored Apr 30, 2020 by Jeff Rasley Committed by GitHub Apr 30, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

requirements.txt requirements.txt +2 -0

No files found.
--- a/requirements.txt
+++ b/requirements.txt
+torch>=1.2
+torchvision>=0.4.0
 pillow==6.2.2
 tqdm
 psutil