def lr_decay(self, episode, episodes):
"""
Decay the actor and critic learning rates.
:param episode: (int) current training episode.
:param episodes: (int) total number of training episodes.
"""
update_linear_schedule(self.actor_optimizer, episode, episodes,
self.lr)
update_linear_schedule(self.critic_optimizer, episode, episodes,
self.critic_lr)
def lr_decay(self, episode, episodes):
update_linear_schedule(self.actor_optimizer, episode, episodes,
self.lr)
update_linear_schedule(self.critic_optimizer, episode, episodes,
self.critic_lr)
def lr_decay(self, episode, episodes):
update_linear_schedule(self.optimizer, episode, episodes, self.lr)