steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-12-06 15:40:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/rl_plotting' into marl_refactor

Browse Source
This commit is contained in:
Chanumask
2024-05-21 11:39:51 +02:00
parent 5e9e59c843 1a8ca9110b
commit 13ea9d25c9
31 changed files with 530 additions and 134 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -18,6 +18,7 @@ from collections import deque
from marl_factory_grid.environment.actions import Noop
from marl_factory_grid.modules import Clean, DoorUse
from marl_factory_grid.utils.plotting.plot_single_runs import plot_action_maps
class Names:
@@ -583,8 +584,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], self.results_path)
@torch.inference_mode(True)
def eval_loop(self, n_episodes, render=False):