Update requirements so PyTorch 1.4 is min, add separate sotabench reqs to force 1.4 and prevent error with 1.5
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
|
||||
torchvision>=0.4.0
|
||||
torch>=1.4.0
|
||||
torchvision>=0.5.0
|
||||
pyyaml
|
||||
|
@ -0,0 +1,5 @@
|
||||
#!/usr/bin/env bash
|
||||
source /workspace/venv/bin/activate
|
||||
|
||||
pip install -r requirements-sotabench.txt
|
||||
|
Loading…
Reference in new issue