correctly rename stddev variable to logvar (log of variance)

alecGraves · May 17, 2019 · 31aebbe · 31aebbe
1 parent 810506b
commit 31aebbe
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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)