コード例 #1
0
 def test_mmd_vectorized(self):
     z_size = 1
     self.do_fit_prior_test(
         True, 2500,
         Trace_MMD(kernel=kernels.RBF(z_size,
                                      lengthscale=torch.sqrt(
                                          torch.tensor(z_size,
                                                       dtype=torch.float))),
                   vectorize_particles=True,
                   num_particles=100))
コード例 #2
0
ファイル: test_inference.py プロジェクト: xidulu/pyro
 def test_mmd_nonvectorized(self):
     z_size = self.loc0.shape[0]
     self.do_fit_prior_test(
         True, 1000,
         Trace_MMD(kernel=kernels.RBF(z_size,
                                      lengthscale=torch.sqrt(
                                          torch.tensor(z_size,
                                                       dtype=torch.float))),
                   vectorize_particles=False,
                   num_particles=100))