diff --git a/src/helios/model/model.py b/src/helios/model/model.py index 52b33e3..ebac310 100644 --- a/src/helios/model/model.py +++ b/src/helios/model/model.py @@ -211,7 +211,6 @@ def on_training_batch_start(self, state: TrainingState) -> None: """ self._loss_items.clear() - @abc.abstractmethod def train_step(self, batch: typing.Any, state: TrainingState) -> None: """ Perform a single training step.