def log_loss(self, episode):
algorithm_logger.warning("Episode: {0} | Critic Loss: {1:.2f}".format(
episode, self.critic_loss))
def save(self, episode):
self.saver.save(self.session, self.save_path)
algorithm_logger.warning("Episode: {} | Saver reach checkpoint.".format(episode))
def save(self, step):
self.saver.save(self.session, self.save_path)
algorithm_logger.warning("Step: {} | Saver reach checkpoint.".format(step + 1))
def log_loss(self, episode):
algorithm_logger.warning("Episode: {0} | Actor Loss: {1:.2f}".format(
episode, self.loss))