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
|
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…
Reference in new issue