fix bug on metrics

mosaicml · Mar 2, 2024 · d3b3a61 · d3b3a61
1 parent d104d16
commit d3b3a61
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/train/train.py b/scripts/train/train.py
@@ -534,7 +534,7 @@ def main(cfg: DictConfig) -> Trainer:
 
     # Now add the eval metrics
     if eval_loader_config is not None and not use_async_eval:
-        train_metrics = model.get_metrics(is_train=True)
+        train_metrics = model.get_metrics(is_train=False)
         evaluators = add_metrics_to_eval_loaders(evaluators,
                                                  list(train_metrics.keys()))