From 3d8db64fa1ccf70333b9cf34d9ed10839b8d6385 Mon Sep 17 00:00:00 2001
From: dlagul <dailiang@iie.ac.cn>
Date: Wed, 28 Oct 2020 16:51:39 +0800
Subject: [PATCH] Update model.py

---
 sdfvae/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sdfvae/model.py b/sdfvae/model.py
index cd01cd7..1571334 100644
--- a/sdfvae/model.py
+++ b/sdfvae/model.py
@@ -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
-- 
GitLab