From 49ebafb509b5156cd29b86ff405c62f21e94cbdf Mon Sep 17 00:00:00 2001 From: lukasugar Date: Tue, 19 May 2026 12:58:36 +0200 Subject: [PATCH] max epochs -> epochs --- asparagus/pipeline/run/linear_probe.py | 2 +- configs/default_linear_probe.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/asparagus/pipeline/run/linear_probe.py b/asparagus/pipeline/run/linear_probe.py index 33a0fcc..090fb05 100644 --- a/asparagus/pipeline/run/linear_probe.py +++ b/asparagus/pipeline/run/linear_probe.py @@ -104,7 +104,7 @@ def main(cfg: DictConfig) -> None: logger=loggers, profiler=None, default_root_dir=path_store.run_dir, - max_epochs=cfg.training.max_epochs, + max_epochs=cfg.training.epochs, check_val_every_n_epoch=cfg.training.check_val_every_n_epoch, limit_train_batches=cfg.training.limit_train_batches, limit_val_batches=cfg.training.limit_val_batches, diff --git a/configs/default_linear_probe.yaml b/configs/default_linear_probe.yaml index 528d083..866c6a2 100644 --- a/configs/default_linear_probe.yaml +++ b/configs/default_linear_probe.yaml @@ -24,7 +24,7 @@ data: training: warmup_epochs: 0 - max_epochs: 15 + epochs: 15 batch_size: 4 limit_train_batches: 1.0 limit_val_batches: 1.0