added rendering of start and target pos. changed file save location to match current run in study out

2025-07-11 23:42:40 +02:00 · 2024-05-16 13:16:53 +02:00
parent cb990445ce
commit 1a8ca9110b
9 changed files with 40 additions and 23 deletions
--- a/marl_factory_grid/algorithms/marl/a2c_dirt.py
+++ b/marl_factory_grid/algorithms/marl/a2c_dirt.py
@ -572,7 +572,7 @@ class A2C:
        if self.cfg[nms.ENV]["save_and_log"]:
            self.create_info_maps(env, used_actions)
            self.save_agent_models()
-            plot_action_maps(env, [self])
+            plot_action_maps(env, [self], self.results_path)

    @torch.inference_mode(True)
    def eval_loop(self, n_episodes, render=False):