fix eval after all (#3445)

mosaicml · Jul 21, 2024 · 6bac335 · 6bac335
1 parent 6b461d0
commit 6bac335
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/composer/core/event.py b/composer/core/event.py
@@ -57,7 +57,7 @@ class Event(StringEnum):
 
                     # <BATCH_END>
 
-                    # <BEFORE_EVAL_ALL>
+                    # <EVAL_BEFORE_ALL>
                     for eval_dataloader in eval_dataloaders:
                         if should_eval(batch=True):
                             # <EVAL_START>
@@ -70,7 +70,7 @@ class Event(StringEnum):
                                 # <EVAL_BATCH_END>
                             # <EVAL_END>
 
-                    # <AFTER_EVAL_ALL>
+                    # <EVAL_AFTER_ALL>
 
                     # <BATCH_CHECKPOINT>
                 # <EPOCH_END>