Fix poolformer group_matcher to merge proj downsample with previous block, support coarse

pull/1256/head
Ross Wightman 3 years ago
parent 37b6920df3
commit 78a32655fa

@ -234,8 +234,8 @@ class PoolFormer(nn.Module):
return dict(
stem=r'^patch_embed', # stem and embed
blocks=[
(r'^network\.(\d+)\.(\d+)', None),
(r'^network\.(\d+)', (0,)),
(r'^network\.(\d+).*\.proj', (99999,)),
(r'^network\.(\d+)', None) if coarse else (r'^network\.(\d+)\.(\d+)', None),
(r'^norm', (99999,))
],
)

Loading…
Cancel
Save