From 3544027d2e692e9d8dcbb5ef842594a53221cbbf Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sat, 29 Oct 2022 20:16:17 +0000 Subject: [PATCH] add tanh activation for pyramid --- models/pyramid_model_optimized.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/pyramid_model_optimized.py b/models/pyramid_model_optimized.py index 92c1d1c..0bec6a4 100644 --- a/models/pyramid_model_optimized.py +++ b/models/pyramid_model_optimized.py @@ -38,7 +38,7 @@ class Pyramid(tf.keras.Model): self.cond_encoder = tf.keras.Sequential([tf.keras.layers.Dense(512, activation='relu'), tf.keras.layers.Dense(256, activation='relu'), tf.keras.layers.Dense(128, activation='relu'), - tf.keras.layers.Dense(config['pyramid']['latent_dim']),]) + tf.keras.layers.Dense(config['pyramid']['latent_dim'], activation='tanh'),]) self.levels = [PyramidLevel({ **config['pyramid']['levels'][i], **{ 'cond_ae': config['cond_ae']} }) for i in range(len(config['pyramid']['levels']))]