fix rtmdet-s cfg and neck (#434)

2023-01-06 11:40:19 +08:00 · 2023-01-06 11:40:19 +08:00 · 5172805dae
parent 28439f102a
commit 5172805dae
2 changed files with 2 additions and 2 deletions
--- a/configs/rtmdet/rtmdet_s_syncbn_fast_8xb32-300e_coco.py
+++ b/configs/rtmdet/rtmdet_s_syncbn_fast_8xb32-300e_coco.py
@ -74,6 +74,6 @@ custom_hooks = [
        priority=49),
    dict(
        type='mmdet.PipelineSwitchHook',
-        switch_epoch=280,
+        switch_epoch=_base_.max_epochs - _base_.stage2_num_epochs,
        switch_pipeline=train_pipeline_stage2)
 ]
--- a/mmyolo/models/necks/cspnext_pafpn.py
+++ b/mmyolo/models/necks/cspnext_pafpn.py
@ -126,7 +126,7 @@ class CSPNeXtPAFPN(BaseYOLONeck):
                conv_cfg=self.conv_cfg,
                norm_cfg=self.norm_cfg,
                act_cfg=self.act_cfg)
-        elif idx == 2:
+        else:
            return nn.Sequential(
                CSPLayer(
                    self.in_channels[idx - 1] * 2,