fixed bug:get GN type in retinanet

lbin · facebook-github-bot · commit 330dd329fc43 · 2022-08-24T11:04:47.000-07:00
Summary: #4369 pytorch/pytorch@4a0f6e6 Pull Request resolved: #4470 Reviewed By: tglik Differential Revision: D38898451 fbshipit-source-id: 44a89aed11cc57a8989b3686a86079627b9dbcb1
diff --git a/detectron2/modeling/meta_arch/retinanet.py b/detectron2/modeling/meta_arch/retinanet.py
@@ -353,7 +353,7 @@ def norm(c):
                 )
 
         else:
-            norm_name = str(type(get_norm(norm, 1)))
+            norm_name = str(type(get_norm(norm, 32)))
             if "BN" in norm_name:
                 logger.warning(
                     f"Shared BatchNorm (type={norm_name}) may not work well in RetinaNetHead."

Original file line number	Diff line number	Diff line change
`@@ -353,7 +353,7 @@ def norm(c):`
`353`	`353`	`)`
`354`	`354`
`355`	`355`	`else:`
`356`		`- norm_name = str(type(get_norm(norm, 1)))`
	`356`	`+ norm_name = str(type(get_norm(norm, 32)))`
`357`	`357`	`if "BN" in norm_name:`
`358`	`358`	`logger.warning(`
`359`	`359`	`f"Shared BatchNorm (type={norm_name}) may not work well in RetinaNetHead."`