Skip to content
Snippets Groups Projects
Commit 6c0fcb72 authored by Lucas Miranda's avatar Lucas Miranda
Browse files

Added support for tensorboard HParams while tuning hyperparameters

parent e588caef
No related branches found
No related tags found
No related merge requests found
......@@ -10,7 +10,6 @@ Simple utility functions used in deepof example scripts. These are not part of t
from datetime import datetime
from kerastuner import BayesianOptimization
from kerastuner import HyperParameters
from kerastuner_tensorboard_logger import TensorBoardLogger
from typing import Tuple, Union, Any, List
import deepof.hypermodels
......@@ -20,8 +19,6 @@ import os
import pickle
import tensorflow as tf
hp = HyperParameters()
def load_hparams(hparams):
"""Loads hyperparameters from a custom dictionary pickled on disc.
......@@ -164,13 +161,10 @@ def tune_search(
predictor=predictor,
)
try:
if "ELBO" in loss and kl_wu > 0:
callbacks.append(hypermodel.kl_warmup_callback)
if "MMD" in loss and mmd_wu > 0:
callbacks.append(hypermodel.mmd_warmup_callback)
except AttributeError:
pass
else:
return False
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment