def on_batch_end(self, runner: "IRunner") -> None:
        """On batch end action: update metric with new batch data
        and log it's value if necessary

        Args:
            runner: current runner
        """
        metrics_inputs = self._get_inputs(runner=runner)
        metrics = self._update_metric(metrics_inputs)
        if self.log_on_batch:
            runner.batch_metrics.update(metrics)
On batch end action: update metric with new batch data
        and log it's value if necessary

        Args:
            runner: current runner
        
catalyst/callbacks/metric.py