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

pull/156/head
Ross Wightman 5 years ago
parent d79ac48626
commit 5966654052

@ -0,0 +1,3 @@
torch==1.4.0
torchvision==0.5.0
pyyaml

@ -1,3 +1,3 @@
torch>=1.2.0 torch>=1.4.0
torchvision>=0.4.0 torchvision>=0.5.0
pyyaml pyyaml

@ -0,0 +1,5 @@
#!/usr/bin/env bash
source /workspace/venv/bin/activate
pip install -r requirements-sotabench.txt
Loading…
Cancel
Save