def load_config():
parser = argparse.ArgumentParser()
parser.add_argument('--config', type=str, required=True)
parser.add_argument('options', default=None, nargs=argparse.REMAINDER)
args = parser.parse_args()
config = get_default_config()
config.merge_from_file(args.config)
config.merge_from_list(args.options)
update_config(config)
config.freeze()
return config
def load_config():
parser = argparse.ArgumentParser()
parser.add_argument('--config', type=str)
parser.add_argument('--resume', type=str, default='')
parser.add_argument('--local_rank', type=int, default=0)
parser.add_argument('options', default=None, nargs=argparse.REMAINDER)
args = parser.parse_args()
config = get_default_config()
if args.config is not None:
config.merge_from_file(args.config)
config.merge_from_list(args.options)
if not torch.cuda.is_available():
config.device = 'cpu'
config.train.dataloader.pin_memory = False
if args.resume != '':
config_path = pathlib.Path(args.resume) / 'config.yaml'
config.merge_from_file(config_path.as_posix())
config.merge_from_list(['train.resume', True])
config.merge_from_list(['train.dist.local_rank', args.local_rank])
config = update_config(config)
config.freeze()
return config
