Commit 52f4351f authored by lucas_miranda's avatar lucas_miranda
Browse files

Changed full model for diagonal model in all variational implementations in models.py

parent 228dd4f0
......@@ -114,7 +114,7 @@ class MultivariateNormalDiag(tfpl.DistributionLambda):
self,
event_size,
convert_to_tensor_fn=tfd.Distribution.sample,
validate_args=False,
validate_args=True,
**kwargs
):
......
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