diff --git a/src/otx/algo/detection/atss.py b/src/otx/algo/detection/atss.py index b9e8326f47d..8800e9a60b8 100644 --- 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, )