From 4cc58ce89149d97edfaef67e7d08c15fab8372df Mon Sep 17 00:00:00 2001 From: Anna Pfohl Date: Wed, 10 Jan 2024 19:07:17 +0000 Subject: [PATCH] eval first --- scripts/train/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/train/train.py b/scripts/train/train.py index 713a5697a5..e3e89472b6 100644 --- 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...')