fix setup.py requirements

inferno-pytorch · Sep 1, 2019 · bcb2b3d · bcb2b3d
1 parent 546c395
commit bcb2b3d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -23,7 +23,9 @@
     "scipy>=0.13.0",
     "h5py",
     "numpy>=1.8",
-    "scikit-image"
+    "scikit-image",
+    "torchvision",
+    "tqdm"
 ]