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)
