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))