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

Update model.py

上级 6d334de4
分支
无相关合并请求
......@@ -181,7 +181,7 @@ class SDFVAE(nn.Module):
if isinstance(m, nn.Conv1d) or isinstance(m, nn.Conv2d) or isinstance(m, nn.ConvTranspose2d) or isinstance(m, nn.Linear):
nn.init.kaiming_normal_(m.weight)
# Time-dependent prior distribution of dynamic latent variables
def sample_d_lstmcell(self, batch_size, random_sampling=True):
d_out = None
d_means = None
......
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册