using pytorch writer now

williamFalcon · Jun 30, 2019 · 493a9c5 · 493a9c5
1 parent 998bee0
commit 493a9c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -14,7 +14,8 @@
         'numpy>=1.13.3',
         'imageio>=2.3.0',
         'tensorboard>=1.14.0',
-        'torch>=1.1.0'
+        'torch>=1.1.0',
+        'future'
     ],
     author_email='will@hacstudios.com',
     url='https://github.com/williamFalcon/test_tube',