From 79121794fe2f2a903fb3975f57dd58289ec1d8b6 Mon Sep 17 00:00:00 2001
From: grimoire <yaoqian@sensetime.com>
Date: Fri, 11 Mar 2022 15:23:17 +0800
Subject: [PATCH] fix l2norm test

---
 tests/test_codebase/test_mmdet/test_mmdet_models.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_codebase/test_mmdet/test_mmdet_models.py b/tests/test_codebase/test_mmdet/test_mmdet_models.py
index 192bf72198..b3c8d90098 100644
--- 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