未验证 提交 349f3dfa 编辑于 作者: dlagul's avatar dlagul 提交者: GitHub
浏览文件

Update model.py

上级 5ab7287f
加载中
加载中
加载中
加载中
+1 −0
原始行号 差异行号 差异行
@@ -262,6 +262,7 @@ class SDFVAE(nn.Module):
            raise ValueError('Unknown encoder and decoder: {}'.format(self.enc_dec))
        return x

    # See https://arxiv.org/pdf/1606.05908.pdf, Page 11, Equation (10) for details.
    def reparameterize(self, mean, logvar, random_sampling=True):
        if random_sampling is True:
            eps = torch.randn_like(logvar)