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))
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))