def on_batch_start(self, state: State) -> None: """Batch start hook. Args: state (State): current state """ state.batch_metrics = defaultdict(None)
def on_batch_end(self, state: State) -> None: """Batch end hook. Args: state (State): current state """ state.batch_metrics = self._process_metrics(state.batch_metrics) for key, value in state.batch_metrics.items(): self.meters[key].add(value)
def on_batch_end(self, state: State): state.batch_metrics = self._process_metrics(state.batch_metrics) for key, value in state.batch_metrics.items(): self.meters[key].add(value)
def on_batch_start(self, state: State): state.batch_metrics = defaultdict(None)