Method log

SwissArmyTransformer/sat/training/utils.py:122–132

Log a group of timers.

(self, names, normalizer=1.0, reset=True)

        return self.timers[name]

    def log(self, names, normalizer=1.0, reset=True):
        """Log a group of timers."""
        assert normalizer > 0.0
        string = 'time (ms)'
        for name in names:
            if name not in self.timers:
                continue
            elapsed_time = self.timers[name].elapsed(
                reset=reset) * 1000.0 / normalizer
            string += ' | {}: {:.2f}'.format(name, elapsed_time)
        print_rank0(string)


def report_memory(name):