def LogIterStats(self, cur_iter, lr):
     """Log the tracked statistics."""
     if (cur_iter % self.LOG_PERIOD == 0 or
             cur_iter == cfg.SOLVER.MAX_ITER - 1):
         stats = self.GetStats(cur_iter, lr)
         log_stats(stats, self.misc_args)
         if self.tblogger:
             self.tb_log_stats(stats, cur_iter)
Exemplo n.º 2
0
def log_training_stats(training_stats, global_step, lr):
    stats = training_stats.GetStats(global_step, lr)
    log_stats(stats, training_stats.misc_args)
    if training_stats.tblogger:
        training_stats.tb_log_stats(stats, global_step)
Exemplo n.º 3
0
def log_training_stats(training_stats, global_step, lr):
    stats = training_stats.GetStats(global_step, lr)
    log_stats(stats, training_stats.misc_args)
Exemplo n.º 4
0
def log_training_stats(training_stats, global_step, lr, input_data, shot):
    stats = training_stats.GetStats(global_step, lr)
    log_stats(stats, training_stats.misc_args)
    if training_stats.tblogger:
        training_stats.tb_log_stats(stats, global_step)
        training_stats.tblogger._add_images(global_step, input_data, shot)