def on_epoch_start(self, state: State) -> None: """Epoch start hook. Args: state (State): current state """ state.epoch_metrics = defaultdict(None)
def on_epoch_start(self, state: State): state.epoch_metrics = defaultdict(None)