def on_batch_start(self, state: State):
"""Batch start hook.
Args:
state (State): current state
"""
state.phase = self.phase_manager.get_phase_name(state)
def on_batch_start(self, state: State):
state.phase = self.phase_manager.get_phase_name(state)