Commit 868b7056 authored by lucas_miranda's avatar lucas_miranda
Browse files

Updated requirements.txt

parent 60cb0f7b
Pipeline #88005 passed with stage
in 23 minutes and 6 seconds
......@@ -11,6 +11,7 @@ keras hypermodels for hyperparameter tuning of deep autoencoders
from kerastuner import HyperModel
import deepof.models
import deepof.model_utils
import numpy as np
import tensorflow_probability as tfp
tfd = tfp.distributions
......@@ -109,7 +110,7 @@ class SEQ_2_SEQ_GMVAE(HyperModel):
bidirectional_merge = hp.Choice(
"bidirectional_merge", values=["sum", "mul", "concat", "ave"]
)
clipvalue = hp.Choice("clipvalue", values=[1.0, None])
clipvalue = hp.Choice("clipvalue", values=[1.0, np.inf])
conv_filters = 160
dense_2 = 120
dense_activation = "relu"
......
Supports Markdown
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