do not log dataset to stdout in each iteration

molmod · Jan 24, 2024 · ff6afad · ff6afad
1 parent 1ba8847
commit ff6afad
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/psiflow/metrics.py b/psiflow/metrics.py
@@ -545,8 +545,7 @@ def log_string(s: str) -> None:
             assert dataset is not None
             inputs = [dataset.data_future, model.evaluate(dataset).data_future]
             dataset_log = log_dataset(inputs=inputs)
-            dataset_log_str = save_dataset_log(dataset_log, path / "dataset.log")
-            log_string(dataset_log_str)
+            save_dataset_log(dataset_log, path / "dataset.log")
         if self.wandb_group is not None:
             #  typically needs a result() from caller
             return to_wandb(  # noqa: F841