fix l2norm test (#232)

2022-03-14 14:45:42 +08:00 · 2022-03-14 14:45:42 +08:00 · 776659a6ce
parent 1f5e670421
commit 776659a6ce
1 changed files with 1 additions and 0 deletions
--- 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