From 37b6920df3f8a271e8286ec34c9cb277c553d16c Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 9 May 2022 10:40:40 -0700 Subject: [PATCH] Fix group_matcher regex for regnet.py --- timm/models/regnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/regnet.py b/timm/models/regnet.py index 3e22bf56..9d1f1f64 100644 --- a/timm/models/regnet.py +++ b/timm/models/regnet.py @@ -458,7 +458,7 @@ class RegNet(nn.Module): 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