From 7203aac212a1b15718cf1a2ca42a9c18de4e7be7 Mon Sep 17 00:00:00 2001
From: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
Date: Thu, 15 Sep 2022 15:11:31 +0800
Subject: [PATCH] [Refactor]Resume do not specify ckpt (#2074)

* [Refactor]Resume do not specify ckpt

* add default false value
---
 tools/train.py | 16 ++++------------
 1 file changed, 4 insertions(+), 12 deletions(-)

diff --git a/tools/train.py b/tools/train.py
index 8c1e5c7da6..172815a9fe 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -18,12 +18,9 @@ def parse_args():
     parser.add_argument('--work-dir', help='the dir to save logs and models')
     parser.add_argument(
         '--resume',
-        nargs='?',
-        type=str,
-        const='auto',
-        help='If specify checkpoint path, resume from it, while if not '
-        'specify, try to auto resume from the latest checkpoint '
-        'in the work directory.')
+        action='store_true',
+        default=False,
+        help='resume from the latest checkpoint in the work_dir automatically')
     parser.add_argument(
         '--amp',
         action='store_true',
@@ -90,12 +87,7 @@ def main():
             cfg.optim_wrapper.loss_scale = 'dynamic'
 
     # resume training
-    if args.resume == 'auto':
-        cfg.resume = True
-        cfg.load_from = None
-    elif args.resume is not None:
-        cfg.resume = True
-        cfg.load_from = args.resume
+    cfg.resume = args.resume
 
     # build the runner from config
     if 'runner_type' not in cfg: