From 31aebbec43ef9c2cb3ac840df1dcf53f4416ca85 Mon Sep 17 00:00:00 2001 From: Alec Graves Date: Fri, 17 May 2019 14:11:47 -0400 Subject: [PATCH] correctly rename stddev variable to logvar (log of variance) --- bvae/models.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/bvae/models.py b/bvae/models.py index 7689c6b..80bb77a 100644 --- a/bvae/models.py +++ b/bvae/models.py @@ -107,11 +107,11 @@ def Build(self): mean = Conv2D(filters=self.latentSize, kernel_size=(1, 1), padding='same')(net) mean = GlobalAveragePooling2D()(mean) - stddev = Conv2D(filters=self.latentSize, kernel_size=(1, 1), + logvar = Conv2D(filters=self.latentSize, kernel_size=(1, 1), padding='same')(net) - stddev = GlobalAveragePooling2D()(stddev) + logvar = GlobalAveragePooling2D()(logvar) - sample = SampleLayer(self.latentConstraints, self.beta)([mean, stddev], training=self.training) + sample = SampleLayer(self.latentConstraints, self.beta)([mean, logvar], training=self.training) return Model(inputs=inLayer, outputs=sample)