diff --git a/train.py b/train.py
index 5b0ff69a..cfe90dea 100644
--- a/train.py
+++ b/train.py
@@ -247,7 +247,7 @@ if __name__ == '__main__':
     parser.add_argument('--weights-path', type=str, default='weights', help='path to store weights')
     parser.add_argument('--resume', action='store_true', help='resume training flag')
     parser.add_argument('--report', action='store_true', help='report TP, FP, FN, P and R per batch (slower)')
-    parser.add_argument('--freeze-darknet53', default=False, help='freeze darknet53.conv.74 layers for first epoch')
+    parser.add_argument('--freeze', action='store_true', help='freeze darknet53.conv.74 layers for first epoche')
     parser.add_argument('--var', type=float, default=0, help='optional test variable')
     opt = parser.parse_args()
     print(opt, end='\n\n')
@@ -265,6 +265,6 @@ if __name__ == '__main__':
         weights_path=opt.weights_path,
         report=opt.report,
         multi_scale=opt.multi_scale,
-        freeze_backbone=opt.freeze_darknet53,
+        freeze_backbone=opt.freeze,
         var=opt.var,
     )