convert all on instantiate

Eclectic-Sheep · Dec 18, 2023 · 2ad65f3 · 2ad65f3
1 parent a8766df
commit 2ad65f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sheeprl/algos/ppo/ppo.py b/sheeprl/algos/ppo/ppo.py
@@ -181,7 +181,7 @@ def main(fabric: Fabric, cfg: Dict[str, Any]):
     )
 
     # Define the optimizer
-    optimizer = hydra.utils.instantiate(cfg.algo.optimizer, params=agent.parameters())
+    optimizer = hydra.utils.instantiate(cfg.algo.optimizer, params=agent.parameters(), _convert_="all")
 
     if fabric.is_global_zero:
         save_configs(cfg, log_dir)