diff --git a/.gitignore b/.gitignore
index 34a68ade0..7d776a671 100644
--- a/.gitignore
+++ b/.gitignore
@@ -14,3 +14,4 @@ experiment-*
 **/.mypy_cache/*
 .mypy_cache/*
 not_tracked_dir/
+.vscode
diff --git a/main.py b/main.py
index e74466f49..e5f9eff80 100644
--- a/main.py
+++ b/main.py
@@ -84,21 +84,21 @@ def get_args_parser():
     parser.add_argument('--coco_panoptic_path', type=str)
     parser.add_argument('--remove_difficult', action='store_true')
 
-    parser.add_argument('--output-dir', default='',
+    parser.add_argument('--output_dir', default='',
                         help='path where to save, empty for no saving')
     parser.add_argument('--device', default='cuda',
                         help='device to use for training / testing')
     parser.add_argument('--seed', default=42, type=int)
     parser.add_argument('--resume', default='', help='resume from checkpoint')
-    parser.add_argument('--start-epoch', default=0, type=int, metavar='N',
+    parser.add_argument('--start_epoch', default=0, type=int, metavar='N',
                         help='start epoch')
     parser.add_argument('--eval', action='store_true')
     parser.add_argument('--num_workers', default=2, type=int)
 
     # distributed training parameters
-    parser.add_argument('--world-size', default=1, type=int,
+    parser.add_argument('--world_size', default=1, type=int,
                         help='number of distributed processes')
-    parser.add_argument('--dist-url', default='env://', help='url used to set up distributed training')
+    parser.add_argument('--dist_url', default='env://', help='url used to set up distributed training')
     return parser