From bb93cbab38014946dd5e70bf5d74d66a43d5ff62 Mon Sep 17 00:00:00 2001
From: Christoph Kowalski <christoph.kowalski@titus-research.eu>
Date: Thu, 12 Sep 2024 15:18:46 +0200
Subject: [PATCH] Added the additional config information to be logged by wandb
to know f.e. which Converter was used
---
.../reinforcement_learning/train_single_agent.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/cooperative_cuisine/reinforcement_learning/train_single_agent.py b/cooperative_cuisine/reinforcement_learning/train_single_agent.py
index 2801a23d..633e6b77 100644
--- a/cooperative_cuisine/reinforcement_learning/train_single_agent.py
+++ b/cooperative_cuisine/reinforcement_learning/train_single_agent.py
@@ -33,7 +33,7 @@ def main(cfg: DictConfig):
vec_env = additional_configs["vec_env"]
number_envs_parallel = config["number_envs_parallel"]
model_class = instantiate(cfg.model.model_type)
- data_to_log=dict(config, **env_info)
+ data_to_log=dict(config, **env_info, **additional_configs)
if vec_env:
env = make_vec_env(lambda: EnvGymWrapper(cfg), n_envs=number_envs_parallel)
else:
--
GitLab