Skip to content
代码片段 群组 项目
未验证 提交 349f3dfa 编辑于 作者: dlagul's avatar dlagul 提交者: GitHub
浏览文件

Update model.py

上级 5ab7287f
无相关合并请求
......@@ -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)
......
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册