transforms.CenterCrop automatically converts image size into a tuple, but transforms.Resize does not. When running models on ImageNet at 384x384 with 1.0 crop_pct, this will result in transforms.Resize(384). Some methods (i.e. ConvNeXt, Swin) implement this with transforms.Resize((384, 384)). As a result their reported scores on ImageNet at 384 are not reproducible with timm's validation script.pull/1383/head
parent
7430a85d07
commit
e91c2181b3
Loading…
Reference in new issue