Skip to content

Commit

Permalink
Formatting.
Browse files Browse the repository at this point in the history
  • Loading branch information
GeorgiosSmyrnis committed Apr 26, 2024
1 parent 1621a94 commit 82f448b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion open_lm/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -331,7 +331,7 @@ def train_one_epoch(
# e.g., saving checkpoints and optmization states that may lead to skipped
# training on restarts.
return False, step

# reset all average meters
losses_m.reset()
if averagers is not None and args.log_avg_model_training_loss:
Expand Down

0 comments on commit 82f448b

Please sign in to comment.