Browse Source

remove useless codes

yjh0410 1 year ago
parent
commit
15a6ec2aca
2 changed files with 2 additions and 3 deletions
  1. 2 2
      models/detectors/rtcdet/rtcdet_basic.py
  2. 0 1
      models/detectors/yolox/loss.py

+ 2 - 2
models/detectors/rtcdet/rtcdet_basic.py

@@ -108,9 +108,9 @@ class RTCBlock(nn.Module):
         super(RTCBlock, self).__init__()
         self.inter_dim = out_dim // 2
         self.input_proj = BasicConv(in_dim, out_dim, kernel_size=1, act_type=act_type, norm_type=norm_type)
-        self.m = nn.Sequential(*(
+        self.m = nn.ModuleList([
             Bottleneck(self.inter_dim, self.inter_dim, 1.0, [3, 3], shortcut, act_type, norm_type, depthwise)
-            for _ in range(num_blocks)))
+            for _ in range(num_blocks)])
         self.output_proj = BasicConv((2 + num_blocks) * self.inter_dim, out_dim, kernel_size=1, act_type=act_type, norm_type=norm_type)
 
     def forward(self, x):

+ 0 - 1
models/detectors/yolox/loss.py

@@ -18,7 +18,6 @@ class Criterion(object):
         self.num_classes = num_classes
         self.max_epoch = args.max_epoch
         self.no_aug_epoch = args.no_aug_epoch
-        self.aux_bbox_loss = False
         # loss weight
         self.loss_obj_weight = cfg['loss_obj_weight']
         self.loss_cls_weight = cfg['loss_cls_weight']