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

Hopefully the last test update for this PR...

Browse Source
This commit is contained in:
Ross Wightman 2021-05-25 11:10:17 -07:00
parent 11ae795e99
commit 99d97e0d67
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/tests.yml vendored
View File

@ -17,8 +17,8 @@ jobs:
matrix: matrix:
os: [ubuntu-latest, macOS-latest] os: [ubuntu-latest, macOS-latest]
python: ['3.8'] python: ['3.8']
torch: ['1.8.0'] torch: ['1.8.1']
torchvision: ['0.9.0'] torchvision: ['0.9.1']
runs-on: ${{ matrix.os }} runs-on: ${{ matrix.os }}
steps: steps:

4
tests/test_models.py
View File

@ -73,7 +73,7 @@ def test_model_forward(model_name, batch_size):
model = create_model(model_name, pretrained=False) model = create_model(model_name, pretrained=False)
model.eval() model.eval()
input_size = _get_input_size(model, TARGET_FWD_SIZE) input_size = _get_input_size(model=model, target=TARGET_FWD_SIZE)
if max(input_size) > MAX_FWD_SIZE: if max(input_size) > MAX_FWD_SIZE:
pytest.skip("Fixed input size model > limit.") pytest.skip("Fixed input size model > limit.")
inputs = torch.randn((batch_size, *input_size)) inputs = torch.randn((batch_size, *input_size))
@ -221,7 +221,7 @@ def test_model_forward_features(model_name, batch_size):
expected_channels = model.feature_info.channels() expected_channels = model.feature_info.channels()
assert len(expected_channels) >= 4 # all models here should have at least 4 feature levels by default, some 5 or 6 assert len(expected_channels) >= 4 # all models here should have at least 4 feature levels by default, some 5 or 6
input_size = _get_input_size(model, TARGET_FFEAT_SIZE) input_size = _get_input_size(model=model, target=TARGET_FFEAT_SIZE)
if max(input_size) > MAX_FFEAT_SIZE: if max(input_size) > MAX_FFEAT_SIZE:
pytest.skip("Fixed input size model > limit.") pytest.skip("Fixed input size model > limit.")