1
Fork 0

refactor(log): use dataclass for config

This commit is contained in:
Tibo De Peuter 2026-04-16 15:18:40 +02:00
parent 37a4b59e04
commit c5b08e817e
Signed by: tdpeuter
SSH key fingerprint: SHA256:u/h/LVoqKF1Iz02uOyxe6hcjmoZASCGV2HM0TG9ZMoU
3 changed files with 17 additions and 28 deletions

View file

@ -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}")