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

param update

Browse Source
This commit is contained in:
Steffen Illium
2021-11-24 17:59:02 +01:00
parent 3833a0a064
commit 3c84ba483b
2 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
reload_agent.py
View File

@@ -50,9 +50,8 @@ if __name__ == '__main__':
rew, done_bool = 0, False
while not done_bool:
if n_agents > 1:
actions = [model.predict(
np.stack([env_state[i][j] for i in range(env_state.shape[0])]),
deterministic=determin)[0] for j, model in enumerate(models)]
actions = [model.predict(env_state[model_idx], deterministic=True)[0]
for model_idx, model in enumerate(models)]
else:
actions = models[0].predict(env_state, deterministic=determin)[0]
env_state, step_r, done_bool, info_obj = env.step(actions)