Update metaformers.py

2 years ago · 42741b7cdb
parent 116345b4d3
commit 42741b7cdb
1 changed files with 1 additions and 1 deletions
--- 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))