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

Try lower max size for non_std default_cfg test

Browse Source
This commit is contained in:
Ross Wightman 2021-06-12 23:31:24 -07:00
parent 4d96165989
commit 0020268d9b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_models.py
View File

@ -174,7 +174,7 @@ def test_model_default_cfgs_non_std(model_name, batch_size):
cfg = model.default_cfg cfg = model.default_cfg
input_size = _get_input_size(model_name=model_name, target=TARGET_FWD_SIZE) input_size = _get_input_size(model_name=model_name, target=TARGET_FWD_SIZE)
if max(input_size) > MAX_FWD_SIZE: if max(input_size) > 320: # FIXME const
pytest.skip("Fixed input size model > limit.") pytest.skip("Fixed input size model > limit.")
input_tensor = torch.randn((batch_size, *input_size)) input_tensor = torch.randn((batch_size, *input_size))