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( return dict(
stem=r'^patch_embed', # stem and embed stem=r'^patch_embed', # stem and embed
blocks=[ blocks=[
(r'^network\.(\d+)\.(\d+)', None), (r'^network\.(\d+).*\.proj', (99999,)),
(r'^network\.(\d+)', (0,)), (r'^network\.(\d+)', None) if coarse else (r'^network\.(\d+)\.(\d+)', None),
(r'^norm', (99999,)) (r'^norm', (99999,))
], ],
) )

Loading…
Cancel
Save