def test_rws_nonreparameterized_vectorized(self): self.do_elbo_test( False, 5000, ReweightedWakeSleep(num_particles=2, vectorize_particles=True, max_plate_nesting=1), )
def test_rws_reparameterized(self): self.do_elbo_test(True, 2500, ReweightedWakeSleep(num_particles=3))
def test_rws_nonreparameterized(self): self.do_elbo_test(False, 5000, ReweightedWakeSleep(num_particles=2))