From f45d2e3386394c76a62abf8c1a62fd0660c60566 Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Thu, 11 Jul 2019 09:20:01 +0200
Subject: [PATCH] #42 run baselines in ci
---
torch_training/bla.py | 18 +++++++++---------
1 file changed, 9 insertions(+), 9 deletions(-)
diff --git a/torch_training/bla.py b/torch_training/bla.py
index 4225d66..f0e7f59 100644
--- a/torch_training/bla.py
+++ b/torch_training/bla.py
@@ -148,15 +148,15 @@ def main(argv):
env_renderer.gl.saveImage("./Images/flatland_frame_{:04d}.bmp".format(frame_step))
frame_step += 1
# print(step)
- # # Action
- # for a in range(env.get_num_agents()):
- # if demo:
- # eps = 0
- # # action = agent.act(np.array(obs[a]), eps=eps)
- # action = agent.act(agent_obs[a], eps=eps)
- # action_prob[action] += 1
- # action_dict.update({a: action})
- # # Environment step
+ # Action
+ for a in range(env.get_num_agents()):
+ if demo:
+ eps = 0
+ # action = agent.act(np.array(obs[a]), eps=eps)
+ action = agent.act(agent_obs[a], eps=eps)
+ action_prob[action] += 1
+ action_dict.update({a: action})
+ # Environment step
#
# next_obs, all_rewards, done, _ = env.step(action_dict)
# # print(all_rewards,action)
--
GitLab