Skip to content

Commit 1b27813

Browse files
committed
Change models with mean 0,0,0 std 1,1,1 from int to float for consistency as mentioned in #1355
1 parent 909705e commit 1b27813

File tree

5 files changed

+8
-8
lines changed

5 files changed

+8
-8
lines changed

timm/models/mlp_mixer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,11 +85,11 @@ def _cfg(url='', **kwargs):
8585
# Mixer ImageNet-21K-P pretraining
8686
mixer_b16_224_miil_in21k=_cfg(
8787
url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/ImageNet_21K_P/models/timm/mixer_b16_224_miil_in21k.pth',
88-
mean=(0, 0, 0), std=(1, 1, 1), crop_pct=0.875, interpolation='bilinear', num_classes=11221,
88+
mean=(0., 0., 0.), std=(1., 1., 1.), crop_pct=0.875, interpolation='bilinear', num_classes=11221,
8989
),
9090
mixer_b16_224_miil=_cfg(
9191
url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/ImageNet_21K_P/models/timm/mixer_b16_224_miil.pth',
92-
mean=(0, 0, 0), std=(1, 1, 1), crop_pct=0.875, interpolation='bilinear',
92+
mean=(0., 0., 0.), std=(1., 1., 1.), crop_pct=0.875, interpolation='bilinear',
9393
),
9494

9595
gmixer_12_224=_cfg(mean=IMAGENET_DEFAULT_MEAN, std=IMAGENET_DEFAULT_STD),

timm/models/mobilenetv3.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,10 +41,10 @@ def _cfg(url='', **kwargs):
4141
interpolation='bicubic',
4242
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_large_100_ra-f55367f5.pth'),
4343
'mobilenetv3_large_100_miil': _cfg(
44-
interpolation='bilinear', mean=(0, 0, 0), std=(1, 1, 1),
44+
interpolation='bilinear', mean=(0., 0., 0.), std=(1., 1., 1.),
4545
url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/ImageNet_21K_P/models/timm/mobilenetv3_large_100_1k_miil_78_0.pth'),
4646
'mobilenetv3_large_100_miil_in21k': _cfg(
47-
interpolation='bilinear', mean=(0, 0, 0), std=(1, 1, 1),
47+
interpolation='bilinear', mean=(0., 0., 0.), std=(1., 1., 1.),
4848
url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/ImageNet_21K_P/models/timm/mobilenetv3_large_100_in21k_miil.pth', num_classes=11221),
4949

5050
'mobilenetv3_small_050': _cfg(

timm/models/mobilevit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def _cfg(url='', **kwargs):
3434
return {
3535
'url': url, 'num_classes': 1000, 'input_size': (3, 256, 256), 'pool_size': (8, 8),
3636
'crop_pct': 0.9, 'interpolation': 'bicubic',
37-
'mean': (0, 0, 0), 'std': (1, 1, 1),
37+
'mean': (0., 0., 0.), 'std': (1., 1., 1.),
3838
'first_conv': 'stem.conv', 'classifier': 'head.fc',
3939
'fixed_input_size': False,
4040
**kwargs

timm/models/tresnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ def _cfg(url='', **kwargs):
2121
return {
2222
'url': url, 'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': (7, 7),
2323
'crop_pct': 0.875, 'interpolation': 'bilinear',
24-
'mean': (0, 0, 0), 'std': (1, 1, 1),
24+
'mean': (0., 0., 0.), 'std': (1., 1., 1.),
2525
'first_conv': 'body.conv1.0', 'classifier': 'head.fc',
2626
**kwargs
2727
}

timm/models/vision_transformer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,12 +163,12 @@ def _cfg(url='', **kwargs):
163163
# ViT ImageNet-21K-P pretraining by MILL
164164
'vit_base_patch16_224_miil_in21k': _cfg(
165165
url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/ImageNet_21K_P/models/timm/vit_base_patch16_224_in21k_miil.pth',
166-
mean=(0, 0, 0), std=(1, 1, 1), crop_pct=0.875, interpolation='bilinear', num_classes=11221,
166+
mean=(0., 0., 0.), std=(1., 1., 1.), crop_pct=0.875, interpolation='bilinear', num_classes=11221,
167167
),
168168
'vit_base_patch16_224_miil': _cfg(
169169
url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/ImageNet_21K_P/models/timm'
170170
'/vit_base_patch16_224_1k_miil_84_4.pth',
171-
mean=(0, 0, 0), std=(1, 1, 1), crop_pct=0.875, interpolation='bilinear',
171+
mean=(0., 0., 0.), std=(1., 1., 1.), crop_pct=0.875, interpolation='bilinear',
172172
),
173173

174174
'vit_base_patch16_rpn_224': _cfg(

0 commit comments

Comments
 (0)