Glenn Jocher
GitHub
3 years ago
1 changed files with 2 additions and 1 deletions
+ 2
- 1
train.py
View File
| if opt.resume: # resume an interrupted run | if opt.resume: # resume an interrupted run | ||||
| ckpt = opt.resume if isinstance(opt.resume, str) else get_latest_run() # specified or most recent path | ckpt = opt.resume if isinstance(opt.resume, str) else get_latest_run() # specified or most recent path | ||||
| assert os.path.isfile(ckpt), 'ERROR: --resume checkpoint does not exist' | assert os.path.isfile(ckpt), 'ERROR: --resume checkpoint does not exist' | ||||
| apriori = opt.global_rank, opt.local_rank | |||||
| with open(Path(ckpt).parent.parent / 'opt.yaml') as f: | with open(Path(ckpt).parent.parent / 'opt.yaml') as f: | ||||
| opt = argparse.Namespace(**yaml.load(f, Loader=yaml.FullLoader)) # replace | opt = argparse.Namespace(**yaml.load(f, Loader=yaml.FullLoader)) # replace | ||||
| opt.cfg, opt.weights, opt.resume = '', ckpt, True | |||||
| opt.cfg, opt.weights, opt.resume, opt.global_rank, opt.local_rank = '', ckpt, True, *apriori # reinstate | |||||
| logger.info('Resuming training from %s' % ckpt) | logger.info('Resuming training from %s' % ckpt) | ||||
| else: | else: | ||||
| # opt.hyp = opt.hyp or ('hyp.finetune.yaml' if opt.weights else 'hyp.scratch.yaml') | # opt.hyp = opt.hyp or ('hyp.finetune.yaml' if opt.weights else 'hyp.scratch.yaml') |
Loading…