def _parse_hydra_args(cfg): # This need to be a separate function which sets globals because hydra doesn't # support returning from its main function global args, config args = cfg check_generic_args(args) config = args.config.to_container()
def parse_classy_args(): """ parses default classy args from sys.argv adding some nice-to-have decorations (e.g. automatically set --device depending on the host type) """ parser = get_parser() args = parser.parse_args() args.config_file = to_abs_path(args.config_file) args.device = "gpu" if torch.cuda.is_available() else "cpu" check_generic_args(args) return args
def hydra_main(cfg): args = cfg check_generic_args(cfg) config = omegaconf.OmegaConf.to_container(cfg.config) main(args, config)
def hydra_main(cfg): args = cfg check_generic_args(cfg) config = cfg.config.to_container() main(args, config)