Skip to content

Commit 3ee118f

Browse files
Expand cls model zoo (#55)
1. expand cls model zoo 2. uniform load_pretrained
1 parent d76d5d7 commit 3ee118f

File tree

125 files changed

+906
-739
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

125 files changed

+906
-739
lines changed

configs/classification/cifar10/r50_b128_300e_jpg.py

-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
# model settings
33
model = dict(
44
type='Classification',
5-
pretrained=None,
65
backbone=dict(
76
type='ResNet',
87
depth=50,

configs/classification/cifar10/swintiny_b64_5e_jpg.py

-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
# model settings
33
model = dict(
44
type='Classification',
5-
pretrained=None,
65
backbone=dict(
76
type='PytorchImageModelWrapper',
87
model_name='swin_tiny_patch4_window7_224',

configs/classification/imagenet/hrnet/hrnetw18_b32x8_100e_jpg.py

-1
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
# model settings
88
model = dict(
99
type='Classification',
10-
pretrained=None,
1110
backbone=dict(type='HRNet', arch='w18', multi_scale_output=True),
1211
neck=dict(type='HRFuseScales', in_channels=(18, 36, 72, 144)),
1312
head=dict(
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
11
_base_ = './hrnetw18_b32x8_100e_jpg.py'
22
# model settings
3-
model = dict(backbone=dict(type='HRNet', arch='w30', multi_scale_output=True))
3+
model = dict(
4+
backbone=dict(type='HRNet', arch='w30', multi_scale_output=True),
5+
neck=dict(type='HRFuseScales', in_channels=(30, 60, 120, 240)))
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
11
_base_ = './hrnetw18_b32x8_100e_jpg.py'
22
# model settings
3-
model = dict(backbone=dict(type='HRNet', arch='w32', multi_scale_output=True))
3+
model = dict(
4+
backbone=dict(type='HRNet', arch='w32', multi_scale_output=True),
5+
neck=dict(type='HRFuseScales', in_channels=(32, 64, 128, 256)))
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
11
_base_ = './hrnetw18_b32x8_100e_jpg.py'
22
# model settings
3-
model = dict(backbone=dict(type='HRNet', arch='w40', multi_scale_output=True))
3+
model = dict(
4+
backbone=dict(type='HRNet', arch='w40', multi_scale_output=True),
5+
neck=dict(type='HRFuseScales', in_channels=(40, 80, 160, 320)))
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
11
_base_ = './hrnetw18_b32x8_100e_jpg.py'
22
# model settings
3-
model = dict(backbone=dict(type='HRNet', arch='w44', multi_scale_output=True))
3+
model = dict(
4+
backbone=dict(type='HRNet', arch='w44', multi_scale_output=True),
5+
neck=dict(type='HRFuseScales', in_channels=(44, 88, 176, 352)))
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
11
_base_ = './hrnetw18_b32x8_100e_jpg.py'
22

33
# model settings
4-
model = dict(backbone=dict(type='HRNet', arch='w48', multi_scale_output=True))
4+
model = dict(
5+
backbone=dict(type='HRNet', arch='w48', multi_scale_output=True),
6+
neck=dict(type='HRFuseScales', in_channels=(48, 96, 192, 384)))
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
11
_base_ = './hrnetw18_b32x8_100e_jpg.py'
22
# model settings
3-
model = dict(backbone=dict(type='HRNet', arch='w64', multi_scale_output=True))
3+
model = dict(
4+
backbone=dict(type='HRNet', arch='w64', multi_scale_output=True),
5+
neck=dict(type='HRFuseScales', in_channels=(64, 128, 256, 512)))

configs/classification/imagenet/resnet/imagenet_resnet101_jpg.py

+1-11
Original file line numberDiff line numberDiff line change
@@ -2,18 +2,8 @@
22
# model settings
33
model = dict(
44
type='Classification',
5-
pretrained=None,
65
backbone=dict(
76
type='ResNet',
87
depth=101,
98
out_indices=[4], # 0: conv-1, x: stage-x
10-
norm_cfg=dict(type='BN')),
11-
head=dict(
12-
type='ClsHead',
13-
with_avg_pool=True,
14-
in_channels=2048,
15-
loss_config=dict(
16-
type='CrossEntropyLossWithLabelSmooth',
17-
label_smooth=0,
18-
),
19-
num_classes=1000))
9+
norm_cfg=dict(type='BN')))
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,8 @@
11
_base_ = './resnet50_b32x8_100e_jpg.py'
22
# model settings
33
model = dict(
4-
type='Classification',
5-
pretrained=None,
64
backbone=dict(
75
type='ResNet',
86
depth=50,
97
out_indices=[4], # 0: conv-1, x: stage-x
10-
norm_cfg=dict(type='BN')),
11-
head=dict(
12-
type='ClsHead',
13-
with_avg_pool=True,
14-
in_channels=2048,
15-
loss_config=dict(
16-
type='CrossEntropyLossWithLabelSmooth',
17-
label_smooth=0,
18-
),
19-
num_classes=1000))
8+
norm_cfg=dict(type='BN')))

configs/classification/imagenet/resnet/resnet50_b32x8_100e_jpg.py

-1
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
# model settings
88
model = dict(
99
type='Classification',
10-
pretrained=None,
1110
backbone=dict(
1211
type='ResNet',
1312
depth=50,

configs/classification/imagenet/resnext/resnext50-32x4d_b32x8_100e_jpg.py

-1
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
# model settings
88
model = dict(
99
type='Classification',
10-
pretrained=None,
1110
backbone=dict(
1211
type='ResNeXt',
1312
depth=50,

configs/classification/imagenet/swint/swin_tiny_patch4_window7_224_b64x16_300e_jpg.py

-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
mode='batch',
1717
label_smoothing=0.1,
1818
num_classes=1000),
19-
pretrained=None,
2019
backbone=dict(
2120
type='PytorchImageModelWrapper',
2221
model_name='swin_tiny_patch4_window7_224',
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='cait_s24_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='cait_xxs24_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='cait_xxs36_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='coat_mini'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='coat_tiny'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='convit_base'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='convit_small'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='convit_tiny'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='convmixer_1024_20_ks9_p14'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='convmixer_1536_20'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='convmixer_768_32'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='convnext_base'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='convnext_large'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='convnext_small'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='convnext_tiny'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='crossvit_base_240'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='crossvit_small_240'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='crossvit_tiny_240'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='deit_base_distilled_patch16_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='deit_base_patch16_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='gmixer_24_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='gmlp_s16_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='levit_128'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='levit_192'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='levit_256'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='mixer_b16_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='mixer_l16_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='mobilevit_s'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='mobilevit_xs'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='mobilevit_xxs'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='jx_nest_base'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='jx_nest_small'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='jx_nest_tiny'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='pit_b_distilled_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='pit_s_distilled_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='poolformer_m36'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='poolformer_m48'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='poolformer_s12'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='poolformer_s24'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='poolformer_s36'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='resmlp_12_distilled_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='resmlp_24_distilled_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='resmlp_36_distilled_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='resmlp_big_24_distilled_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='sequencer2d_l'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='sequencer2d_m'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='sequencer2d_s'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='shuffletrans_base_p4_w7_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='shuffletrans_small_p4_w7_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='shuffletrans_tiny_p4_w7_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='dynamic_swin_small_p4_w7_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='dynamic_swin_tiny_p4_w7_224'))
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
_base_ = '../timm_config.py'
2+
3+
# model settings
4+
model = dict(backbone=dict(model_name='swin_base_patch4_window7_224'))

0 commit comments

Comments
 (0)