Skip to content

Commit

Permalink
[BugFix] missing model
Browse files Browse the repository at this point in the history
  • Loading branch information
fedebotu committed Jul 19, 2024
1 parent 02f3dfa commit af7c37b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion run.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ def run(cfg: DictConfig) -> Tuple[dict, dict]:
callbacks: List[Callback] = utils.instantiate_callbacks(cfg.get("callbacks"))

log.info("Instantiating loggers...")
logger: List[Logger] = utils.instantiate_loggers(cfg.get("logger"))
logger: List[Logger] = utils.instantiate_loggers(cfg.get("logger"), model=model)

log.info("Instantiating trainer...")
trainer: RL4COTrainer = hydra.utils.instantiate(
Expand Down

0 comments on commit af7c37b

Please sign in to comment.