debug YOLOvx-Nano on COCO

models/detectors/yolovx/yolovx.py

@@ -29,7 +29,7 @@ class YOLOvx(nn.Module):
         # ---------------------- Basic Parameters ----------------------
         self.cfg = cfg
         self.device = device
-        self.stride = cfg['stride']
+        self.strides = cfg['stride']
         self.num_classes = num_classes
         self.trainable = trainable
         self.conf_thresh = conf_thresh
@@ -52,12 +52,12 @@ class YOLOvx(nn.Module):
 
         ## ----------- Heads -----------
         self.det_heads = build_det_head(
-            cfg, self.fpn_dims, self.head_dim, num_classes, num_levels=len(self.stride))
+            cfg, self.fpn_dims, self.head_dim, num_classes, num_levels=len(self.strides))
 
         ## ----------- Preds -----------
         self.pred_layers = build_pred_layer(
             self.det_heads.cls_head_dim, self.det_heads.reg_head_dim,
-            self.stride, num_classes, num_coords=4, num_levels=len(self.stride))
+            self.strides, num_classes, num_coords=4, num_levels=len(self.strides))
 
 
     ## post-process

models/detectors/yolovx/yolovx_pred.py

@@ -62,7 +62,7 @@ class MultiLevelPredLayer(nn.Module):
         self.num_coords = num_coords
         self.num_levels = num_levels
 
-        ## ----------- Network Parameters -----------
+        # ----------- Network Parameters -----------
         self.multi_level_preds = nn.ModuleList(
             [SingleLevelPredLayer(
                 cls_dim,