mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Update metaformers.py

Browse Source
This commit is contained in:
Fredo Guan 2023-01-08 09:37:49 -08:00
parent 7aa3459caf
commit 944fd549c4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/models/metaformers.py
View File

@ -650,7 +650,7 @@ class MetaFormer(nn.Module):
stages = nn.ModuleList() # each stage consists of multiple metaformer blocks stages = nn.ModuleList() # each stage consists of multiple metaformer blocks
cur = 0 cur = 0
for i in range(num_stage): for i in range(num_stage):
stage = nn.Sequential(OrderedDict[ stage = nn.Sequential(OrderedDict([
('downsample', downsample_layers[i]), ('downsample', downsample_layers[i]),
('blocks', nn.Sequential(*[MetaFormerBlock( ('blocks', nn.Sequential(*[MetaFormerBlock(
dim=dims[i], dim=dims[i],
@ -661,7 +661,7 @@ class MetaFormer(nn.Module):
layer_scale_init_value=layer_scale_init_values[i], layer_scale_init_value=layer_scale_init_values[i],
res_scale_init_value=res_scale_init_values[i] res_scale_init_value=res_scale_init_values[i]
) for j in range(depths[i])]) ) for j in range(depths[i])])
)] )])
) )
stages.append(stage) stages.append(stage)
cur += depths[i] cur += depths[i]