From 7d50637396f045e01c86b3789be63067dcc07e6b Mon Sep 17 00:00:00 2001 From: Lars Bogner Date: Thu, 27 Nov 2025 10:25:04 +0100 Subject: [PATCH] Increase epochs for mode 3 --- Code/python/notebooks/hyperparameter_scan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Code/python/notebooks/hyperparameter_scan.py b/Code/python/notebooks/hyperparameter_scan.py index 0c9ce34..db82768 100644 --- a/Code/python/notebooks/hyperparameter_scan.py +++ b/Code/python/notebooks/hyperparameter_scan.py @@ -306,7 +306,7 @@ def run_mode_3(): print(f"Training {base_name} warm={warm} pred={pred} altitude_weight={altitude_weight}") - loss = train_one_model(model, train_ds, val_ds, warm // step, pred // step, altitude_weight=altitude_weight, batch_size=64) + loss = train_one_model(model, train_ds, val_ds, warm // step, pred // step, altitude_weight=altitude_weight, batch_size=64, epochs=250) torch.save(model.state_dict(), f"{base_name}_wu{warm}_ps{pred}_aw{altitude_weight}.pt") results.append((base_name, warm, pred, altitude_weight, loss))