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)