[Fix] set optional arg a default value (open-mmlab#483)

* optional default value * resolve comments Co-authored-by: dongchunyu.vendor <[email protected]> (cherry picked from commit ba641c3)
lvhan028 · Jun 7, 2022 · a932bf9 · a932bf9
1 parent cdaa16b
commit a932bf9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tools/deploy.py b/tools/deploy.py
@@ -1,6 +1,7 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 import argparse
 import logging
+import os
 import os.path as osp
 from functools import partial
 
@@ -27,7 +28,10 @@ def parse_args():
     parser.add_argument('img', help='image used to convert model model')
     parser.add_argument(
         '--test-img', default=None, help='image used to test model')
-    parser.add_argument('--work-dir', help='the dir to save logs and models')
+    parser.add_argument(
+        '--work-dir',
+        default=os.getcwd(),
+        help='the dir to save logs and models')
     parser.add_argument(
         '--calib-dataset-cfg',
         help='dataset config path used to calibrate in int8 mode. If not \