def sample_prior(self, prob_decode=False):
    tree_vec = create_var(torch.randn(1, (self.latent_size // 2)), False)
    mol_vec = create_var(torch.randn(1, (self.latent_size // 2)), False)
    return self.decode(tree_vec, mol_vec, prob_decode)
def sample_prior(self, prob_decode=False):
    tree_vec = create_var(torch.randn(1, (self.latent_size // 2)), False)
    mol_vec = create_var(torch.randn(1, (self.latent_size // 2)), False)
    return self.decode(tree_vec, mol_vec, prob_decode)