diff --git a/run.py b/run.py
index 7506140..2bc9107 100644
--- a/run.py
+++ b/run.py
@@ -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(