mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Ross Wightman 2020-05-24 13:28:03 -07:00
parent d79ac48626
commit 5966654052
3 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
requirements-sotabench.txt Normal file
View File

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

4
requirements.txt
View File

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

5
sotabench_setup.sh Executable file
View File

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