From c5d7ce8422c353271faed371a2cfcf2cfb912a7b Mon Sep 17 00:00:00 2001 From: Mateusz Klimaszewski <mk.klimaszewski@gmail.com> Date: Thu, 6 Aug 2020 10:28:57 +0200 Subject: [PATCH] Remove torchvision, fix torch version. --- setup.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 328d05d..03c533a 100644 --- a/setup.py +++ b/setup.py @@ -10,8 +10,7 @@ REQUIREMENTS = [ 'requests==2.23.0', 'overrides==3.0.0', 'tensorboard==2.1.0', - 'torch>=1.5.0', - 'torchvision>=0.6.0', + 'torch>=1.5.0,<1.6.0', 'tqdm==4.43.0', 'transformers==2.9.1', 'urllib3==1.24.2', -- GitLab