diff --git a/cv/detection/yolov7/pytorch/train.py b/cv/detection/yolov7/pytorch/train.py
index 86c7e48d5ac214ad4a4c0a63b924d7ece429211c..f1dfd4d9189330dce83479cb2996e3c7f4da9117 100644
--- a/cv/detection/yolov7/pytorch/train.py
+++ b/cv/detection/yolov7/pytorch/train.py
@@ -1,3 +1,6 @@
+# Copyright (c) 2024, Shanghai Iluvatar CoreX Semiconductor Co., Ltd.
+# All Rights Reserved.
+
 import argparse
 import logging
 import math
@@ -547,7 +550,7 @@ if __name__ == '__main__':
     parser.add_argument('--single-cls', action='store_true', help='train multi-class data as single-class')
     parser.add_argument('--adam', action='store_true', help='use torch.optim.Adam() optimizer')
     parser.add_argument('--sync-bn', action='store_true', help='use SyncBatchNorm, only available in DDP mode')
-    parser.add_argument('--local_rank', type=int, default=-1, help='DDP parameter, do not modify')
+    parser.add_argument('--local_rank', '--local-rank', type=int, default=-1, help='DDP parameter, do not modify')
     parser.add_argument('--workers', type=int, default=8, help='maximum number of dataloader workers')
     parser.add_argument('--project', default='runs/train', help='save to project/name')
     parser.add_argument('--entity', default=None, help='W&B entity')