eval first

mosaicml · Jan 10, 2024 · 4cc58ce · 4cc58ce
1 parent c2ef669
commit 4cc58ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/train/train.py b/scripts/train/train.py
@@ -626,7 +626,7 @@ def main(cfg: DictConfig) -> Trainer:
     gc.collect()
 
     # Eval first if requested
-    if eval_first and trainer.state.timestamp.batch.value == 0 and not use_async_eval:
+    if eval_first and trainer.state.timestamp.batch.value == 0:
         trainer.eval()
 
     log.info('Starting training...')