Commit 4be486bd authored by lucas_miranda's avatar lucas_miranda
Browse files

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

parent 8af352be
Pipeline #94309 canceled with stage
in 1 minute and 41 seconds
......@@ -429,7 +429,7 @@ def autoencoder_fitting(
callbacks=callbacks_,
)
if not os.path.exists("trained_weights"):
if not os.path.exists(os.path.join(output_path, "trained_weights")):
os.makedirs("trained_weights")
if save_weights:
......@@ -536,7 +536,9 @@ def tune_search(
hpt_params = {
"hypermodel": hypermodel,
"executions_per_trial": n_replicas,
"logger": TensorBoardLogger(metrics=["val_mae"], logdir="./logs/hparams"),
"logger": TensorBoardLogger(
metrics=["val_mae"], logdir=os.path.join(outpath, "logged_hparams")
),
"objective": "val_mae",
"project_name": project_name,
"seed": 42,
......@@ -550,7 +552,7 @@ def tune_search(
),
max_epochs=35,
hyperband_iterations=hypertun_trials,
factor=2,
factor=3,
**hpt_params
)
else:
......
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