steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2026-01-15 23:41:39 +01:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Chanumask
2024-05-16 13:16:53 +02:00
parent cb990445ce
commit 1a8ca9110b
9 changed files with 40 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
marl_factory_grid/algorithms/marl/a2c_dirt.py
View File

@@ -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):