diff --git a/configs/barlowtwins/benchmarks/resnet50_8xb32-linear-coslr-100e_in1k.py b/configs/barlowtwins/benchmarks/resnet50_8xb32-linear-coslr-100e_in1k.py index 7b365127b9b..2f4e4f574ff 100644 --- a/configs/barlowtwins/benchmarks/resnet50_8xb32-linear-coslr-100e_in1k.py +++ b/configs/barlowtwins/benchmarks/resnet50_8xb32-linear-coslr-100e_in1k.py @@ -5,7 +5,10 @@ '../../_base_/default_runtime.py', ] -model = dict(backbone=dict(frozen_stages=4)) +model = dict( + backbone=dict( + frozen_stages=4, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.'))) # runtime settings default_hooks = dict( diff --git a/configs/beit/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py b/configs/beit/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py index 2a467e658d3..dbab34f6e08 100644 --- a/configs/beit/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py +++ b/configs/beit/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py @@ -23,7 +23,8 @@ out_type='avg_featmap', use_abs_pos_emb=False, use_rel_pos_bias=True, - use_shared_rel_pos_bias=False), + use_shared_rel_pos_bias=False, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=None, head=dict( type='LinearClsHead', diff --git a/configs/beitv2/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py b/configs/beitv2/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py index ee8fabeca18..a2c55a706b3 100644 --- a/configs/beitv2/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py +++ b/configs/beitv2/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py @@ -17,7 +17,8 @@ out_type='avg_featmap', use_abs_pos_emb=False, use_rel_pos_bias=True, - use_shared_rel_pos_bias=False), + use_shared_rel_pos_bias=False, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=None, head=dict( type='LinearClsHead', diff --git a/configs/byol/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py b/configs/byol/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py index 973d2b2215e..2b5074c082b 100644 --- a/configs/byol/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py +++ b/configs/byol/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py @@ -5,7 +5,10 @@ '../../_base_/default_runtime.py', ] -model = dict(backbone=dict(frozen_stages=4)) +model = dict( + backbone=dict( + frozen_stages=4, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.'))) # dataset summary train_dataloader = dict(batch_size=512) diff --git a/configs/cae/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py b/configs/cae/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py index 67f2d373516..e7083ce80a8 100644 --- a/configs/cae/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py +++ b/configs/cae/benchmarks/beit-base-p16_8xb128-coslr-100e_in1k.py @@ -74,7 +74,7 @@ use_abs_pos_emb=True, use_rel_pos_bias=True, use_shared_rel_pos_bias=False, - init_cfg=dict(type='Pretrained', checkpoint='')), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=None, head=dict( type='LinearClsHead', diff --git a/configs/densecl/benchmarks/resnet50_8xb32-linear-steplr-100e_in1k.py b/configs/densecl/benchmarks/resnet50_8xb32-linear-steplr-100e_in1k.py index f7d82d1ebf9..37795d9c866 100644 --- a/configs/densecl/benchmarks/resnet50_8xb32-linear-steplr-100e_in1k.py +++ b/configs/densecl/benchmarks/resnet50_8xb32-linear-steplr-100e_in1k.py @@ -5,7 +5,10 @@ '../../_base_/default_runtime.py', ] -model = dict(backbone=dict(frozen_stages=4)) +model = dict( + backbone=dict( + frozen_stages=4, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.'))) # optimizer optim_wrapper = dict( diff --git a/configs/eva/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py b/configs/eva/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py index a90185de487..e8a3f4983ac 100644 --- a/configs/eva/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py +++ b/configs/eva/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py @@ -58,7 +58,7 @@ drop_path_rate=0.1, out_type='avg_featmap', final_norm=False, - init_cfg=dict(type='Pretrained', checkpoint='')), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=None, head=dict( type='LinearClsHead', diff --git a/configs/eva/benchmarks/vit-base-p16_8xb2048-linear-coslr-100e_in1k.py b/configs/eva/benchmarks/vit-base-p16_8xb2048-linear-coslr-100e_in1k.py index 20db6e56a47..0b7333ca475 100644 --- a/configs/eva/benchmarks/vit-base-p16_8xb2048-linear-coslr-100e_in1k.py +++ b/configs/eva/benchmarks/vit-base-p16_8xb2048-linear-coslr-100e_in1k.py @@ -19,7 +19,7 @@ frozen_stages=12, out_type='cls_token', final_norm=True, - init_cfg=dict(type='Pretrained', checkpoint='')), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=dict(type='ClsBatchNormNeck', input_features=768), head=dict( type='VisionTransformerClsHead', diff --git a/configs/mae/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py b/configs/mae/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py index 511815623f4..4cf9ca11347 100644 --- a/configs/mae/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py +++ b/configs/mae/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py @@ -57,7 +57,7 @@ drop_path_rate=0.1, out_type='avg_featmap', final_norm=False, - init_cfg=dict(type='Pretrained', checkpoint='')), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=None, head=dict( type='LinearClsHead', diff --git a/configs/mae/benchmarks/vit-base-p16_8xb2048-linear-coslr-90e_in1k.py b/configs/mae/benchmarks/vit-base-p16_8xb2048-linear-coslr-90e_in1k.py index 39b14e57bd6..b0545c99d00 100644 --- a/configs/mae/benchmarks/vit-base-p16_8xb2048-linear-coslr-90e_in1k.py +++ b/configs/mae/benchmarks/vit-base-p16_8xb2048-linear-coslr-90e_in1k.py @@ -20,7 +20,7 @@ frozen_stages=12, out_type='cls_token', final_norm=True, - init_cfg=dict(type='Pretrained', checkpoint='')), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=dict(type='ClsBatchNormNeck', input_features=768), head=dict( type='VisionTransformerClsHead', diff --git a/configs/mae/benchmarks/vit-huge-p14_32xb8-coslr-50e_in1k-448px.py b/configs/mae/benchmarks/vit-huge-p14_32xb8-coslr-50e_in1k-448px.py index 9ac9dbb49f8..60046b48d49 100644 --- a/configs/mae/benchmarks/vit-huge-p14_32xb8-coslr-50e_in1k-448px.py +++ b/configs/mae/benchmarks/vit-huge-p14_32xb8-coslr-50e_in1k-448px.py @@ -59,7 +59,7 @@ drop_path_rate=0.3, # set to 0.3 out_type='avg_featmap', final_norm=False, - init_cfg=dict(type='Pretrained', checkpoint='')), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=None, head=dict( type='LinearClsHead', diff --git a/configs/mae/benchmarks/vit-huge-p14_8xb128-coslr-50e_in1k.py b/configs/mae/benchmarks/vit-huge-p14_8xb128-coslr-50e_in1k.py index 71ba3696db2..2a9ff51890b 100644 --- a/configs/mae/benchmarks/vit-huge-p14_8xb128-coslr-50e_in1k.py +++ b/configs/mae/benchmarks/vit-huge-p14_8xb128-coslr-50e_in1k.py @@ -58,7 +58,7 @@ drop_path_rate=0.3, # set to 0.3 out_type='avg_featmap', final_norm=False, - init_cfg=dict(type='Pretrained', checkpoint='')), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=None, head=dict( type='LinearClsHead', diff --git a/configs/mae/benchmarks/vit-large-p16_8xb128-coslr-50e_in1k.py b/configs/mae/benchmarks/vit-large-p16_8xb128-coslr-50e_in1k.py index 9ad6909af50..ae86b40b8a2 100644 --- a/configs/mae/benchmarks/vit-large-p16_8xb128-coslr-50e_in1k.py +++ b/configs/mae/benchmarks/vit-large-p16_8xb128-coslr-50e_in1k.py @@ -58,7 +58,7 @@ drop_path_rate=0.2, # set to 0.2 out_type='avg_featmap', final_norm=False, - init_cfg=dict(type='Pretrained', checkpoint='')), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=None, head=dict( type='LinearClsHead', diff --git a/configs/mae/benchmarks/vit-large-p16_8xb2048-linear-coslr-90e_in1k.py b/configs/mae/benchmarks/vit-large-p16_8xb2048-linear-coslr-90e_in1k.py index 8b363fd5d06..c89518141c1 100644 --- a/configs/mae/benchmarks/vit-large-p16_8xb2048-linear-coslr-90e_in1k.py +++ b/configs/mae/benchmarks/vit-large-p16_8xb2048-linear-coslr-90e_in1k.py @@ -20,7 +20,7 @@ frozen_stages=24, out_type='cls_token', final_norm=True, - init_cfg=dict(type='Pretrained', checkpoint='')), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=dict(type='ClsBatchNormNeck', input_features=1024), head=dict( type='VisionTransformerClsHead', diff --git a/configs/maskfeat/benchmarks/vit-base-p16_8xb256-coslr-100e_in1k.py b/configs/maskfeat/benchmarks/vit-base-p16_8xb256-coslr-100e_in1k.py index 29722b34ed5..5a7620b46b4 100644 --- a/configs/maskfeat/benchmarks/vit-base-p16_8xb256-coslr-100e_in1k.py +++ b/configs/maskfeat/benchmarks/vit-base-p16_8xb256-coslr-100e_in1k.py @@ -56,7 +56,7 @@ drop_path_rate=0.1, out_type='avg_featmap', final_norm=False, - init_cfg=dict(type='Pretrained', checkpoint='')), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=None, head=dict( type='LinearClsHead', diff --git a/configs/milan/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py b/configs/milan/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py index a90185de487..e8a3f4983ac 100644 --- a/configs/milan/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py +++ b/configs/milan/benchmarks/vit-base-p16_8xb128-coslr-100e_in1k.py @@ -58,7 +58,7 @@ drop_path_rate=0.1, out_type='avg_featmap', final_norm=False, - init_cfg=dict(type='Pretrained', checkpoint='')), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=None, head=dict( type='LinearClsHead', diff --git a/configs/milan/benchmarks/vit-base-p16_8xb2048-linear-coslr-100e_in1k.py b/configs/milan/benchmarks/vit-base-p16_8xb2048-linear-coslr-100e_in1k.py index 20db6e56a47..0b7333ca475 100644 --- a/configs/milan/benchmarks/vit-base-p16_8xb2048-linear-coslr-100e_in1k.py +++ b/configs/milan/benchmarks/vit-base-p16_8xb2048-linear-coslr-100e_in1k.py @@ -19,7 +19,7 @@ frozen_stages=12, out_type='cls_token', final_norm=True, - init_cfg=dict(type='Pretrained', checkpoint='')), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=dict(type='ClsBatchNormNeck', input_features=768), head=dict( type='VisionTransformerClsHead', diff --git a/configs/mixmim/benchmarks/mixmim-base_8xb128-coslr-100e_in1k.py b/configs/mixmim/benchmarks/mixmim-base_8xb128-coslr-100e_in1k.py index 01e93205e23..c48ee3b8b64 100644 --- a/configs/mixmim/benchmarks/mixmim-base_8xb128-coslr-100e_in1k.py +++ b/configs/mixmim/benchmarks/mixmim-base_8xb128-coslr-100e_in1k.py @@ -86,6 +86,10 @@ ) test_dataloader = val_dataloader +model = dict( + backbone=dict( + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.'))) + # optimizer optim_wrapper = dict( type='OptimWrapper', diff --git a/configs/mocov2/benchmarks/resnet50_8xb32-linear-steplr-100e_in1k.py b/configs/mocov2/benchmarks/resnet50_8xb32-linear-steplr-100e_in1k.py index f7d82d1ebf9..37795d9c866 100644 --- a/configs/mocov2/benchmarks/resnet50_8xb32-linear-steplr-100e_in1k.py +++ b/configs/mocov2/benchmarks/resnet50_8xb32-linear-steplr-100e_in1k.py @@ -5,7 +5,10 @@ '../../_base_/default_runtime.py', ] -model = dict(backbone=dict(frozen_stages=4)) +model = dict( + backbone=dict( + frozen_stages=4, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.'))) # optimizer optim_wrapper = dict( diff --git a/configs/mocov3/benchmarks/resnet50_8xb128-linear-coslr-90e_in1k.py b/configs/mocov3/benchmarks/resnet50_8xb128-linear-coslr-90e_in1k.py index 4af5e1705ec..4d0b202b0f6 100644 --- a/configs/mocov3/benchmarks/resnet50_8xb128-linear-coslr-90e_in1k.py +++ b/configs/mocov3/benchmarks/resnet50_8xb128-linear-coslr-90e_in1k.py @@ -8,7 +8,11 @@ # dataset settings train_dataloader = dict(batch_size=128) -model = dict(backbone=dict(frozen_stages=4, norm_eval=True)) +model = dict( + backbone=dict( + frozen_stages=4, + norm_eval=True, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.'))) # optimizer optim_wrapper = dict( diff --git a/configs/mocov3/benchmarks/vit-base-p16_8xb128-linear-coslr-90e_in1k.py b/configs/mocov3/benchmarks/vit-base-p16_8xb128-linear-coslr-90e_in1k.py index 7a6af081849..91509fc05d6 100644 --- a/configs/mocov3/benchmarks/vit-base-p16_8xb128-linear-coslr-90e_in1k.py +++ b/configs/mocov3/benchmarks/vit-base-p16_8xb128-linear-coslr-90e_in1k.py @@ -16,7 +16,8 @@ patch_size=16, stop_grad_conv1=True, frozen_stages=12, - norm_eval=True), + norm_eval=True, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), head=dict( type='VisionTransformerClsHead', num_classes=1000, diff --git a/configs/mocov3/benchmarks/vit-base-p16_8xb64-coslr-150e_in1k.py b/configs/mocov3/benchmarks/vit-base-p16_8xb64-coslr-150e_in1k.py index 9678ff52104..f3d074f6ed9 100644 --- a/configs/mocov3/benchmarks/vit-base-p16_8xb64-coslr-150e_in1k.py +++ b/configs/mocov3/benchmarks/vit-base-p16_8xb64-coslr-150e_in1k.py @@ -12,7 +12,7 @@ img_size=224, patch_size=16, drop_path_rate=0.1, - ), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=None, head=dict( type='VisionTransformerClsHead', diff --git a/configs/mocov3/benchmarks/vit-large-p16_8xb64-coslr-100e_in1k.py b/configs/mocov3/benchmarks/vit-large-p16_8xb64-coslr-100e_in1k.py index 29247c7ccd1..46d7f48299e 100644 --- a/configs/mocov3/benchmarks/vit-large-p16_8xb64-coslr-100e_in1k.py +++ b/configs/mocov3/benchmarks/vit-large-p16_8xb64-coslr-100e_in1k.py @@ -12,7 +12,7 @@ img_size=224, patch_size=16, drop_path_rate=0.5, - ), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), neck=None, head=dict( type='VisionTransformerClsHead', diff --git a/configs/mocov3/benchmarks/vit-small-p16_8xb128-linear-coslr-90e_in1k.py b/configs/mocov3/benchmarks/vit-small-p16_8xb128-linear-coslr-90e_in1k.py index 04d0d223412..0c1ffa19726 100644 --- a/configs/mocov3/benchmarks/vit-small-p16_8xb128-linear-coslr-90e_in1k.py +++ b/configs/mocov3/benchmarks/vit-small-p16_8xb128-linear-coslr-90e_in1k.py @@ -16,7 +16,8 @@ patch_size=16, stop_grad_conv1=True, frozen_stages=12, - norm_eval=True), + norm_eval=True, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), head=dict( type='VisionTransformerClsHead', num_classes=1000, diff --git a/configs/simclr/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py b/configs/simclr/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py index 973d2b2215e..2b5074c082b 100644 --- a/configs/simclr/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py +++ b/configs/simclr/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py @@ -5,7 +5,10 @@ '../../_base_/default_runtime.py', ] -model = dict(backbone=dict(frozen_stages=4)) +model = dict( + backbone=dict( + frozen_stages=4, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.'))) # dataset summary train_dataloader = dict(batch_size=512) diff --git a/configs/simmim/benchmarks/swin-base-w6_8xb256-coslr-100e_in1k-192px.py b/configs/simmim/benchmarks/swin-base-w6_8xb256-coslr-100e_in1k-192px.py index 8da8c5d66b6..47c4fa1ccfa 100644 --- a/configs/simmim/benchmarks/swin-base-w6_8xb256-coslr-100e_in1k-192px.py +++ b/configs/simmim/benchmarks/swin-base-w6_8xb256-coslr-100e_in1k-192px.py @@ -9,7 +9,8 @@ backbone=dict( img_size=192, drop_path_rate=0.1, - stage_cfgs=dict(block_cfgs=dict(window_size=6)))) + stage_cfgs=dict(block_cfgs=dict(window_size=6)), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.'))) # optimizer settings optim_wrapper = dict( diff --git a/configs/simmim/benchmarks/swin-base-w7_8xb256-coslr-100e_in1k.py b/configs/simmim/benchmarks/swin-base-w7_8xb256-coslr-100e_in1k.py index dcf62273c8f..f7325f03d6b 100644 --- a/configs/simmim/benchmarks/swin-base-w7_8xb256-coslr-100e_in1k.py +++ b/configs/simmim/benchmarks/swin-base-w7_8xb256-coslr-100e_in1k.py @@ -52,7 +52,8 @@ backbone=dict( img_size=224, drop_path_rate=0.1, - stage_cfgs=dict(block_cfgs=dict(window_size=7)))) + stage_cfgs=dict(block_cfgs=dict(window_size=7)), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.'))) # optimizer settings optim_wrapper = dict( diff --git a/configs/simmim/benchmarks/swin-large-w14_8xb256-coslr-100e_in1k.py b/configs/simmim/benchmarks/swin-large-w14_8xb256-coslr-100e_in1k.py index d1f005187f4..a6eafd84d3c 100644 --- a/configs/simmim/benchmarks/swin-large-w14_8xb256-coslr-100e_in1k.py +++ b/configs/simmim/benchmarks/swin-large-w14_8xb256-coslr-100e_in1k.py @@ -54,7 +54,8 @@ img_size=224, drop_path_rate=0.2, stage_cfgs=dict(block_cfgs=dict(window_size=14)), - pad_small_map=True), + pad_small_map=True, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), head=dict(in_channels=1536)) # optimizer settings diff --git a/configs/simsiam/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py b/configs/simsiam/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py index 973d2b2215e..2b5074c082b 100644 --- a/configs/simsiam/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py +++ b/configs/simsiam/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py @@ -5,7 +5,10 @@ '../../_base_/default_runtime.py', ] -model = dict(backbone=dict(frozen_stages=4)) +model = dict( + backbone=dict( + frozen_stages=4, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.'))) # dataset summary train_dataloader = dict(batch_size=512) diff --git a/configs/spark/benchmarks/convnextv2-tiny_8xb256-coslr-300e_in1k.py b/configs/spark/benchmarks/convnextv2-tiny_8xb256-coslr-300e_in1k.py index 53eff37a1c8..95ef81f16a8 100644 --- a/configs/spark/benchmarks/convnextv2-tiny_8xb256-coslr-300e_in1k.py +++ b/configs/spark/benchmarks/convnextv2-tiny_8xb256-coslr-300e_in1k.py @@ -55,7 +55,7 @@ drop_path_rate=0.1, layer_scale_init_value=0., use_grn=True, - ), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), head=dict( type='LinearClsHead', num_classes=1000, diff --git a/configs/spark/benchmarks/resnet50_8xb256-coslr-300e_in1k.py b/configs/spark/benchmarks/resnet50_8xb256-coslr-300e_in1k.py index c82bcf6b8af..5d7527ce2a5 100644 --- a/configs/spark/benchmarks/resnet50_8xb256-coslr-300e_in1k.py +++ b/configs/spark/benchmarks/resnet50_8xb256-coslr-300e_in1k.py @@ -48,7 +48,7 @@ backbone=dict( norm_cfg=dict(type='SyncBN', requires_grad=True), drop_path_rate=0.05, - ), + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.')), head=dict( loss=dict( type='LabelSmoothLoss', label_smooth_val=0.1, use_sigmoid=True)), diff --git a/configs/swav/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py b/configs/swav/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py index 973d2b2215e..2b5074c082b 100644 --- a/configs/swav/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py +++ b/configs/swav/benchmarks/resnet50_8xb512-linear-coslr-90e_in1k.py @@ -5,7 +5,10 @@ '../../_base_/default_runtime.py', ] -model = dict(backbone=dict(frozen_stages=4)) +model = dict( + backbone=dict( + frozen_stages=4, + init_cfg=dict(type='Pretrained', checkpoint='', prefix='backbone.'))) # dataset summary train_dataloader = dict(batch_size=512)