refactor(log): use dataclass for config
This commit is contained in:
parent
37a4b59e04
commit
c5b08e817e
3 changed files with 17 additions and 28 deletions
|
|
@ -36,13 +36,9 @@ def main(dict_cfg: DictConfig):
|
|||
cfg_dict = OmegaConf.to_container(dict_cfg, resolve=True, throw_on_missing=True)
|
||||
init_logger(
|
||||
run_name=run_name,
|
||||
config=cfg_dict,
|
||||
project_name=config.logging.wandb_project_name,
|
||||
entity=config.logging.wandb_entity,
|
||||
full_config=cfg_dict,
|
||||
logging_cfg=config.logging,
|
||||
base_dir=os.path.dirname(run_dir),
|
||||
use_wandb=config.logging.track,
|
||||
upload_final_model=config.logging.upload_final_model,
|
||||
upload_checkpoints=config.logging.upload_checkpoints,
|
||||
)
|
||||
logger = get_logger()
|
||||
logger.info(f"Hydra-initialized run: {run_name}")
|
||||
|
|
|
|||
Reference in a new issue