Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-05-31 13:05:40 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add checkpoints from the ablation study.

Browse Source 
PiperOrigin-RevId: 328023346
This commit is contained in:
Florent Altché
2020-08-23 14:26:26 +01:00
committed by Diego de Las Casas
parent 22c3daff19
commit 8457046b2c
33 changed files with 397 additions and 363 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hierarchical_probabilistic_unet/model.py
+2 -2
View File
@@ -145,8 +145,8 @@ class _HierarchicalCore(snt.AbstractModule):
regularizers=self._regularizers,
)(decoder_features)
mu = mu_logsigma[Ellipsis, :latent_dim]
logsigma = mu_logsigma[Ellipsis, latent_dim:]
mu = mu_logsigma[..., :latent_dim]
logsigma = mu_logsigma[..., latent_dim:]
dist = tfd.MultivariateNormalDiag(loc=mu, scale_diag=tf.exp(logsigma))
distributions.append(dist)