fix l2norm test

open-mmlab · Mar 11, 2022 · 7912179 · 7912179
1 parent 120f4ac
commit 7912179
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
@@ -125,6 +125,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