Method log_tensorboard

LanguageNetwork/BERT/models/stats.py:116–120

display statistics to tensorboard

(self, prefix, writer, learning_rate, step)

        sys.stdout.flush()

    def log_tensorboard(self, prefix, writer, learning_rate, step):
        """ display statistics to tensorboard """
        t = self.elapsed_time()
        writer.add_scalar(prefix + "/xent", self.xent(), step)
        writer.add_scalar(prefix + "/lr", learning_rate, step)