update

train.py

@@ -285,7 +285,7 @@ def train(hyp,  # path/to/hyp.yaml or hyp dictionary
     if opt.auxotaloss: # https://github.com/iscyy/yoloair
         compute_loss_ota = ComputeLossAuxOTA(model)  # init loss class
         compute_loss = ComputeLoss(model) 
-    elif opt.otaloss:
+    elif opt.otaloss == 'yolov7':
         compute_loss_ota = ComputeLossOTA_v7(model)
         compute_loss = ComputeLoss(model)
     if loss_category is None:
@@ -509,7 +509,7 @@ def parse_opt(known=False):
 
     parser.add_argument('--loss', type=str, default='origin', help='')
     parser.add_argument('--auxotaloss', action='store_true', help='')
-    parser.add_argument('--otaloss', action='store_true', help='use yolov7')
+    parser.add_argument('--otaloss', type=str, default='origin', help='use yolov7 or yolox')
     parser.add_argument('--batch-size', type=int, default=2, help='total batch size for all GPUs, -1 for autobatch')
     parser.add_argument('--imgsz', '--img', '--img-size', type=int, default=160, help='train, val image size (pixels)')
     parser.add_argument('--rect', action='store_true', help='rectangular training')