From 0c584a6a139ee6b9ed2762a4a28256cfcd83e525 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Thu, 8 Aug 2019 14:05:03 -0400
Subject: [PATCH] updated support for 1.2.0 (#80)

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 73240e2f9f..81706665a3 100755
--- a/setup.py
+++ b/setup.py
@@ -29,9 +29,9 @@ setup(
     keywords=['deep learning', 'pytorch', 'AI'],
     python_requires='>=3.6',
     install_requires=[
-        'torch==1.1.0',
+        'torch==1.2.0',
         'tqdm',
-        'test-tube>=0.6.7.6',
+        'test-tube==0.6.8',
         'pandas>=0.20.3',
     ],
     classifiers=[