Made more changes

2024-04-16 22:07:53 +02:00
parent 937b6abc0b
commit 0edcc91e65
12 changed files with 214 additions and 36 deletions
--- a/src/trainers/quantile_trainer.py
+++ b/src/trainers/quantile_trainer.py
@@ -192,9 +192,10 @@ class AutoRegressiveQuantileTrainer(AutoRegressiveTrainer):
                        test_loader=dataloader,
                        initial_penalty=900,
                        target_charge_cycles=283,
-                        learning_rate=2,
+                        initial_learning_rate=5,
                        max_iterations=100,
                        tolerance=1,
+                        iteration=epoch,
                    )
                )

@@ -823,7 +824,7 @@ class NonAutoRegressiveQuantileRegression(Trainer):
                    test_loader=dataloader,
                    initial_penalty=500,
                    target_charge_cycles=283,
-                    learning_rate=2,
+                    initial_learning_rate=2,
                    max_iterations=100,
                    tolerance=1,
                )