diff --git a/deepof/train_model.py b/deepof/train_model.py
index 11ea1ed9bd69597a9f2a02aeab4074f46e1ad8b6..c6ce62a9b88b09ccf54c427f01ce65b7748e7346 100644
--- a/deepof/train_model.py
+++ b/deepof/train_model.py
@@ -434,7 +434,7 @@ else:
         project_name="{}-based_{}_{}".format(input_type, hyp, tune.capitalize()),
         callbacks=[
             tensorboard_callback,
-            #onecycle,
+            onecycle,
             tf.keras.callbacks.EarlyStopping(
                 "val_mae", patience=5, restore_best_weights=True
             ),
diff --git a/deepof/train_utils.py b/deepof/train_utils.py
index 9cf77df2390ed2bca47640727d522814ce92fb2a..c75e8c56786ee4115cf3032262d131e8041e9912 100644
--- a/deepof/train_utils.py
+++ b/deepof/train_utils.py
@@ -196,6 +196,7 @@ def tune_search(
             directory="HyperBandx_{}_{}".format(loss, str(date.today())),
             max_epochs=hypertun_trials,
             hyperband_iterations=3,
+            factor=2,
             **hpt_params
         )
     else: