Fix group_matcher regex for regnet.py

rwightman · rwightman · commit 37b6920df3f8 · 2022-05-09T10:40:40.000-07:00
diff --git a/timm/models/regnet.py b/timm/models/regnet.py
@@ -458,7 +458,7 @@ def _get_stage_args(self, cfg: RegNetCfg, default_stride=2, output_stride=32, dr
     def group_matcher(self, coarse=False):
         return dict(
             stem=r'^stem',
-            blocks=r'^stages\.(\d+)' if coarse else r'^stages\.(\d+)\.blocks\.(\d+)',
+            blocks=r'^s(\d+)' if coarse else r'^s(\d+)\.b(\d+)',
         )
 
     @torch.jit.ignore

Original file line number	Diff line number	Diff line change
`@@ -458,7 +458,7 @@ def _get_stage_args(self, cfg: RegNetCfg, default_stride=2, output_stride=32, dr`
`458`	`458`	`def group_matcher(self, coarse=False):`
`459`	`459`	`return dict(`
`460`	`460`	`stem=r'^stem',`
`461`		`- blocks=r'^stages\.(\d+)' if coarse else r'^stages\.(\d+)\.blocks\.(\d+)',`
	`461`	`+ blocks=r'^s(\d+)' if coarse else r'^s(\d+)\.b(\d+)',`
`462`	`462`	`)`
`463`	`463`
`464`	`464`	`@torch.jit.ignore`