fix eval after all (mosaicml#3445)

mvpatel2000 · Jul 21, 2024 · 8c653d5 · 8c653d5
1 parent 767bb82
commit 8c653d5
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>