def objective(trial: optuna.trial): trial, trial_config = _process_trial_config(trial, config.copy()) runner: ConfigRunner = get_config_runner(expdir=Path(args.expdir), config=trial_config) # @TODO: here we need better solution. runner._trial = trial # noqa: WPS437 if get_rank() <= 0: dump_environment(logdir=runner.logdir, config=config, configs_path=args.configs) dump_code(expdir=args.expdir, logdir=runner.logdir) runner.run() return trial.best_score
def config_main(args, unknown_args): """Yaml config catalyst-dl run entry point.""" args, config = parse_args_uargs(args, unknown_args) set_global_seed(args.seed) prepare_cudnn(args.deterministic, args.benchmark) runner: ConfigRunner = get_config_runner(expdir=args.expdir, config=config) if get_rank() <= 0: dump_environment(logdir=runner.logdir, config=config, configs_path=args.configs) dump_code(expdir=args.expdir, logdir=runner.logdir) runner.run()
def main(cfg: DictConfig): """ Hydra config catalyst-dl run entry point Args: cfg: (DictConfig) configuration """ cfg = prepare_hydra_config(cfg) set_global_seed(cfg.args.seed) prepare_cudnn(cfg.args.deterministic, cfg.args.benchmark) import_module(hydra.utils.to_absolute_path(cfg.args.expdir)) runner = hydra.utils.instantiate(cfg.runner, cfg=cfg) if get_rank() <= 0: dump_environment(logdir=runner.logdir, config=cfg) dump_code(expdir=hydra.utils.to_absolute_path(cfg.args.expdir), logdir=runner.logdir) runner.run()