Merge commit 'refs/pull/232/head' of https://github.com/open-mmlab/mm…

…deploy into sb/mo_args
open-mmlab · Mar 14, 2022 · 84853a9 · 84853a9
2 parents 0b52e8f + 7912179
commit 84853a9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_codebase/test_mmdet/test_mmdet_models.py b/tests/test_codebase/test_mmdet/test_mmdet_models.py
@@ -126,6 +126,7 @@ def get_l2norm_forward_model():
     """L2Norm Neck Config."""
     from mmdet.models.necks.ssd_neck import L2Norm
     model = L2Norm(16)
+    torch.nn.init.uniform_(model.weight)
 
     model.requires_grad_(False)
     return model