diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index a7404cf9..656f6294 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -601,7 +601,7 @@ class MetaFormer(nn.Module): res_scale_init_value=res_scale_init_values[i], ) for j in range(depths[i])] ) - self.stages.append(stage) + stages.append(stage) cur += depths[i] self.stages = nn.Sequential(zip(*downsample_layers, *stages))