Skip to content
Snippets Groups Projects
Commit 7365be1a authored by Egli Adrian (IT-SCI-API-PFI)'s avatar Egli Adrian (IT-SCI-API-PFI)
Browse files

Policy updated

parent 7387e9a5
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
reinforcement_learning/multi_agent_training.py
+ 1
− 0
View file @ 7365be1a
......@@ -431,6 +431,7 @@ def train_agent(train_params, train_env_params, eval_env_params, obs_params):
writer.add_scalar("timer/learn", learn_timer.get(), episode_idx)
writer.add_scalar("timer/preproc", preproc_timer.get(), episode_idx)
writer.add_scalar("timer/total", training_timer.get_current(), episode_idx)
writer.flush()
def format_action_prob(action_probs):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment