diff --git a/timm/models/poolformer.py b/timm/models/poolformer.py
index 3cf6b1a3..17d657b0 100644
--- a/timm/models/poolformer.py
+++ b/timm/models/poolformer.py
@@ -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,))
             ],
         )