debug optim

francesco-vaselli · Feb 25, 2024 · e0ad7b3 · e0ad7b3
1 parent 8c03a04
commit e0ad7b3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/models/param_scan.py b/src/models/param_scan.py
@@ -295,7 +295,7 @@ def train(
         tuner = kt.BayesianOptimization(
             lambda hp: build_cnn_model(hp, target=target),
             objective="val_loss",
-            max_trials=1000,
+            max_trials=100,
             num_initial_points=None,
             alpha=0.0001,
             beta=2.6,
@@ -306,7 +306,7 @@ def train(
         tuner = kt.BayesianOptimization(
             lambda hp: build_rnn_model(hp, target=target),
             objective="val_loss",
-            max_trials=1000,
+            max_trials=100,
             num_initial_points=None,
             alpha=0.0001,
             beta=2.6,
@@ -317,7 +317,7 @@ def train(
         tuner = kt.BayesianOptimization(
             lambda hp: build_transformer_model(hp, target=target),
             objective="val_loss",
-            max_trials=1000,
+            max_trials=100,
             num_initial_points=None,
             alpha=0.0001,
             beta=2.6,