Commit 8af352be authored by lucas_miranda's avatar lucas_miranda
Browse files

Added latent regularization control to deepof.data.coordinates.deep_unsupervised_embedding()

parent 7e0a1405
Pipeline #94294 canceled with stage
in 18 minutes and 48 seconds
......@@ -411,7 +411,7 @@ else:
start_epoch=max(kl_wu, mmd_wu),
),
],
n_replicas=3,
n_replicas=1,
n_epochs=30,
outpath=output_path,
)
......
......@@ -545,7 +545,9 @@ def tune_search(
if hpt_type == "hyperband":
tuner = Hyperband(
directory=os.path.join(outpath, "HyperBandx_{}_{}".format(loss, str(date.today()))),
directory=os.path.join(
outpath, "HyperBandx_{}_{}".format(loss, str(date.today()))
),
max_epochs=35,
hyperband_iterations=hypertun_trials,
factor=2,
......@@ -553,7 +555,9 @@ def tune_search(
)
else:
tuner = BayesianOptimization(
directory=os.path.join(outpath, "BayOpt_{}_{}".format(loss, str(date.today()))),
directory=os.path.join(
outpath, "BayOpt_{}_{}".format(loss, str(date.today()))
),
max_trials=hypertun_trials,
**hpt_params
)
......@@ -577,7 +581,7 @@ def tune_search(
epochs=n_epochs,
validation_data=(Xvals, yvals),
verbose=1,
batch_size=32,
batch_size=64,
callbacks=callbacks,
)
......
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