2b77a13f3870712284122f2a5411c7d4b8db82e6,yarll/agents/sac.py,SAC,learn,#SAC#,253

Before Change

                        summary.value.add(tag="global/Episode_length",
                                          simple_value=float(episode_length))
                        summary.value.add(tag="global/Reward",
                                          simple_value=float(episode_reward))
                        self.summary_writer.add_summary(summary, episode)
                        self.summary_writer.flush()
                        break

After Change

                        self.total_episodes += 1
                        tf.summary.scalar("env/Episode_length", episode_length, self.total_steps)
                        tf.summary.scalar("env/Reward", episode_reward, self.total_steps)
                        tf.summary.scalar("env/N_episodes", self.total_episodes, step=self.total_steps)
                        break

class ActorNetwork(Model):
    def __init__(self, n_hidden_layers, n_hidden_units, n_actions, logprob_epsilon):

Instances