steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-06-18 18:52:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

monitor now returning info objects

Browse Source
This commit is contained in:
steffen-illium
2021-05-31 13:58:24 +02:00
parent 7b4e60b0aa
commit 403d38dc24
7 changed files with 61 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
environments/logging/training.py
View File

@ -32,7 +32,7 @@ class TraningMonitor(BaseCallback):
df.to_csv(self.filepath, mode='a', header=False)
def _on_step(self) -> bool:
for idx, done in np.ndenumerate(self.locals['dones']):
for idx, done in np.ndenumerate(self.locals.get('dones', [])):
idx = idx[0]
# self.values[self.num_timesteps].update(**{f'reward_env_{idx}': self.locals['rewards'][idx]})
self.rewards[idx] += self.locals['rewards'][idx]