Disable slow metrics during train/val by default

pluskal-lab · Feb 10, 2025 · 6bf9ee7 · 6bf9ee7
1 parent 12344ab
commit 6bf9ee7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/massspecgym/models/base.py b/massspecgym/models/base.py
@@ -28,8 +28,8 @@ def __init__(
         self,
         lr: float = 1e-4,
         weight_decay: float = 0.0,
-        log_only_loss_at_stages: T.Sequence[Stage | str] = (),
-        no_mces_metrics_at_stages: T.Sequence[Stage | str] = (),
+        log_only_loss_at_stages: T.Sequence[Stage | str] = (Stage.TRAIN),
+        no_mces_metrics_at_stages: T.Sequence[Stage | str] = (Stage.VAL),
         bootstrap_metrics: bool = False,
         df_test_path: T.Optional[str | Path] = None,
         **kwargs