Skip to content

Commit

Permalink
Assert that ELBO did not become NaN during each step of inference of …
Browse files Browse the repository at this point in the history
…gCNV (#6186)
  • Loading branch information
asmirnov239 authored Oct 28, 2019
1 parent 9012c74 commit 22eb06c
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -392,6 +392,7 @@ def _update_continuous_posteriors(self) -> bool:
try:
for _ in progress_bar:
loss = self.continuous_model_step_func() / self.elbo_normalization_factor
assert not np.isnan(loss), "The optimization step for ELBO update returned a NaN"
self.i_advi += 1

try:
Expand Down

0 comments on commit 22eb06c

Please sign in to comment.