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