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

Test input size for efficientnet_v2s was wrong in last results run

Browse Source
This commit is contained in:
Ross Wightman 2021-04-17 16:17:41 -07:00
parent a04427d8ce
commit e8a64fb881
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/efficientnet.py
View File

@ -156,7 +156,7 @@ default_cfgs = {
'efficientnet_v2s': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_v2s_ra2-b265c1ba.pth',
input_size=(3, 224, 224), test_size=(3, 288, 288), pool_size=(7, 7), crop_pct=1.0), # FIXME WIP
input_size=(3, 224, 224), test_input_size=(3, 320, 320), pool_size=(7, 7), crop_pct=1.0), # FIXME WIP
'tf_efficientnet_b0': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b0_aa-827b6e33.pth',