Skip to content

Commit 2f888cd

Browse files
committed
change tensorboard step to global step count
1 parent ad85178 commit 2f888cd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,8 +73,8 @@ def test(rank, args, shared_model, gl_step_cnt):
7373
time.strftime("%Hh %Mm %Ss",
7474
time.gmtime(passed_time)),
7575
reward_sum, episode_length))
76-
tb.log_value('steps_second', global_step_count / passed_time, local_episode_num)
77-
tb.log_value('reward', reward_sum, local_episode_num)
76+
tb.log_value('steps_second', global_step_count / passed_time, global_step_count)
77+
tb.log_value('reward', reward_sum, global_step_count)
7878

7979
reward_sum = 0
8080
episode_length = 0

0 commit comments

Comments
 (0)