Commit 67949328 authored by lucas_miranda's avatar lucas_miranda
Browse files

Changed epochs default for model training

parent ce0bad51
......@@ -882,7 +882,7 @@ class coordinates:
preprocessed_object: Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray],
batch_size: int = 256,
encoding_size: int = 4,
epochs: int = 35,
epochs: int = 50,
hparams: dict = None,
kl_warmup: int = 0,
log_history: bool = True,
......
......@@ -600,7 +600,7 @@ def tune_search(
directory=os.path.join(
outpath, "HyperBandx_{}_{}".format(loss, str(date.today()))
),
max_epochs=50,
max_epochs=30,
hyperband_iterations=hypertun_trials,
factor=3,
**hpt_params
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment