Skip to content

Commit

Permalink
[brief] Make the train_step function not abstract.
Browse files Browse the repository at this point in the history
[detailed]
- This actually makes the model base class more flexible because it
  means that now the user can create models that isolate each individual
  step (test/train/validate) as opposed to one that has to cover
  training even when it's not needed.
  • Loading branch information
marovira committed May 23, 2024
1 parent bc5f6aa commit d492b3e
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion src/helios/model/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down

0 comments on commit d492b3e

Please sign in to comment.