rm TODO

delphi-suite · Mar 8, 2024 · 7213aa4 · 7213aa4
1 parent 7cb4ca7
commit 7213aa4
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/delphi/train/train_step.py b/src/delphi/train/train_step.py
@@ -55,7 +55,6 @@ def train_step(
         if losses["val"] < model_training_state.best_val_loss:
             model_training_state.best_val_loss = float(losses["val"])
             new_best_val_loss = True
-        # TODO: refactor EvalData to use ModelTrainingState
         eval_data = EvalData(
             tokens_per_iter=iteration_params.tokens_per_iter,
             losses=losses,