From c81bfcf3874ba0e24352c95ac3b994a64de332b4 Mon Sep 17 00:00:00 2001 From: George Smyrnis Date: Fri, 26 Apr 2024 12:06:00 -0500 Subject: [PATCH] Formatting. --- open_lm/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/open_lm/train.py b/open_lm/train.py index 83cd22d3..01072abd 100644 --- a/open_lm/train.py +++ b/open_lm/train.py @@ -333,7 +333,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: