Remove weight initialization arguments for ATSS (#3483)

remove weight init for atss
openvinotoolkit · May 10, 2024 · eaedc34 · eaedc34
1 parent 9cfbc99
commit eaedc34
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/otx/algo/detection/atss.py b/src/otx/algo/detection/atss.py
@@ -288,7 +288,6 @@ def _build_model(self, num_classes: int) -> SingleStageDetector:
             in_channels=64,
             stacked_convs=4,
             feat_channels=64,
-            init_cfg={"type": "Xavier", "layer": "Conv2d", "distribution": "uniform"},
             train_cfg=train_cfg,
             test_cfg=test_cfg,
         )
@@ -361,7 +360,6 @@ def _build_model(self, num_classes: int) -> SingleStageDetector:
             in_channels=256,
             stacked_convs=4,
             feat_channels=256,
-            init_cfg={"type": "Xavier", "layer": "Conv2d", "distribution": "uniform"},
             train_cfg=train_cfg,
             test_cfg=test_cfg,
         )