added init to test folder

Lightning-AI · Jul 25, 2019 · caf8745 · caf8745
1 parent ff0459d
commit caf8745
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -5,4 +5,5 @@ scikit-learn==0.20.2
 tqdm==4.32.1
 twine==1.13.0
 numpy==1.16.4
-torch>=1.1.0
+torch>=1.1.0
+torchvision==0.3.0