예제 #1
0
 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_json_stats(stats)
         return (cur_iter, stats["loss"])
예제 #2
0
 def LogIterStats(self, cur_iter, lr):
     """Log the tracked statistics."""
     num_iter_per_epoch = self.model.roi_data_loader.get_num_iter_per_epoch()
     if (cur_iter % self.LOG_PERIOD == 0 or
             cur_iter == cfg.SOLVER.MAX_ITER * num_iter_per_epoch - 1):
         stats = self.GetStats(cur_iter, lr)
         log_json_stats(stats)
    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_json_stats(stats)

            if self.tblogger:
                self.tb_log_stats(stats, cur_iter)
예제 #4
0
    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_json_stats(stats)

        if self.mem is not None:
            mem_sorted = sorted(self.mem.items(), key=lambda d: d[1])
            print(mem_sorted)
예제 #5
0
 def LogIterStats(self, cur_iter, lr, viz, win_accuracy_cls, win_loss,
                  win_loss_bbox, win_loss_cls):
     """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)
         try:
             log_json_stats(stats, viz, win_accuracy_cls, win_loss,
                            win_loss_bbox, win_loss_cls)
         except:
             log_json_stats(stats)
예제 #6
0
 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_json_stats(stats)
예제 #7
0
 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)
         self.log.append(log_json_stats(stats))