chunk_content: r.Sequential),
   ('stages.0.blocks.0', timm.models.metaformer.MetaFormerBlock),
   ('stages.0.blocks.0.norm1', timm.layers.norm.GroupNorm1),
   ('stages.0.blocks.0.token_mixer', timm.models.metaformer.Pooling),
   ('stages.0.blocks.0.token_mixer.pool', torch.nn.modules.pooling.AvgPool2d),
   ('stages.0.blocks.0.drop_path1', torch.nn.modules.linear.Identity),
   ('stages.0.blocks.0.layer_scale1', timm.models.metaformer.Scale),
   ('stages.0.blocks.0.res_scal