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

Minor test change

Browse Source
This commit is contained in:
Ross Wightman 2020-05-21 15:51:47 -07:00
parent afb6bd0669
commit 3873ea710e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_models.py
View File

@ -66,5 +66,5 @@ def test_model_default_cfgs(model_name, batch_size):
input_size = tuple([min(x, 448) for x in input_size])
outputs = model.forward_features(torch.randn((batch_size, *input_size)))
assert outputs.shape[-1] == pool_size[-1] and outputs.shape[-2] == pool_size[-2]
assert any([k.startswith(cfg['classifier']) for k in state_dict.keys()]), f'{classifier} not in model params'
assert any([k.startswith(cfg['first_conv']) for k in state_dict.keys()]), f'{first_conv} not in model params'
assert any([k.startswith(classifier) for k in state_dict.keys()]), f'{classifier} not in model params'
assert any([k.startswith(first_conv) for k in state_dict.keys()]), f'{first_conv} not in model params'