Skip to content

Commit

Permalink
[Refactor]Resume do not specify ckpt (#2074)
Browse files Browse the repository at this point in the history
* [Refactor]Resume do not specify ckpt

* add default false value
  • Loading branch information
MeowZheng authored Sep 15, 2022
1 parent 0097dfb commit 3388cfd
Showing 1 changed file with 4 additions and 12 deletions.
16 changes: 4 additions & 12 deletions tools/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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',
Expand Down Expand Up @@ -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:
Expand Down

0 comments on commit 3388cfd

Please sign in to comment.