Method update

fourm/utils/logger.py:210–219

(self, metrics)

            self.step += 1

    def update(self, metrics):
        log_dict = dict()
        for k, v in metrics.items():
            if v is None:
                continue
            if isinstance(v, torch.Tensor):
                v = v.item()
            log_dict[k] = v

        self.wandb_safe_log(log_dict, step=self.step)

    def flush(self):
        pass