Increase epochs for mode 3

2025-11-27 10:25:04 +01:00
parent 51094364b2
commit 7d50637396
1 changed files with 1 additions and 1 deletions
@@ -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))