浏览代码

debug yolovx_n on coco

yjh0410 2 年之前
父节点
当前提交
7a4e5959a4
共有 1 个文件被更改,包括 0 次插入1 次删除
  1. 0 1
      models/detectors/yolovx/yolovx.py

+ 0 - 1
models/detectors/yolovx/yolovx.py

@@ -29,7 +29,6 @@ class YOLOvx(nn.Module):
         self.cfg = cfg
         self.device = device
         self.stride = cfg['stride']
-        self.reg_max = cfg['reg_max']
         self.num_classes = num_classes
         self.trainable = trainable
         self.conf_thresh = conf_thresh