Update requirements so PyTorch 1.4 is min, add separate sotabench reqs to force 1.4 and prevent error with 1.5

rwightman · rwightman · commit 5966654052b2 · 2020-05-24T13:28:03.000-07:00
diff --git a/requirements-sotabench.txt b/requirements-sotabench.txt
@@ -0,0 +1,3 @@
+torch==1.4.0
+torchvision==0.5.0
+pyyaml
diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,3 @@
-torch>=1.2.0
-torchvision>=0.4.0
+torch>=1.4.0
+torchvision>=0.5.0
 pyyaml
diff --git a/sotabench_setup.sh b/sotabench_setup.sh
@@ -0,0 +1,5 @@
+#!/usr/bin/env bash
+source /workspace/venv/bin/activate
+
+pip install -r requirements-sotabench.txt
+
