From b2904db84aec721d249e2f67a3f4eb99a77e4b64 Mon Sep 17 00:00:00 2001
From: lucas_miranda <lucasmiranda42@gmail.com>
Date: Tue, 1 Dec 2020 14:43:52 +0100
Subject: [PATCH] Trials without onecycle
---
deepof/train_model.py | 2 +-
deepof/train_utils.py | 1 +
2 files changed, 2 insertions(+), 1 deletion(-)
diff --git a/deepof/train_model.py b/deepof/train_model.py
index 11ea1ed9..c6ce62a9 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 9cf77df2..c75e8c56 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:
--
GitLab