Added support for pytorch 1.2

requirments.txt

@@ -1,6 +1,6 @@
-python=3.6.3
-pytorch=0.3.0
-torchvision=0.2.0
+python=3.7.3
+pytorch=1.2.0
+torchvision=0.4.0
 numpy=1.15.4
 tensorboardx=1.4
 matplotlib=3.0.1