Changed dtype_hint to tf.float64 for numerical stability

pymc-devs · Jul 27, 2020 · 5137396 · 5137396
1 parent 109eda8
commit 5137396
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc4/variational/approximations.py b/pymc4/variational/approximations.py
@@ -106,7 +106,7 @@ class MeanField(Approximation):
     def _build_posterior(self):
         flattened_shape = self.order.size
         dtype = dtype_util.common_dtype(
-            self.state.all_unobserved_values.values(), dtype_hint=tf.float32
+            self.state.all_unobserved_values.values(), dtype_hint=tf.float64
         )
         loc = tf.Variable(tf.random.normal([flattened_shape], dtype=dtype), name="mu")
         cov_param = tfp.util.TransformedVariable(