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

Update model.py

上级 7e5c98fe

sdfvae/model.py

+3 −3
原始行号 差异行号 差异行
@@ -252,7 +252,7 @@ class SDFVAE(nn.Module): 
        return x

    

 

    def decoder_logvar(self, sdh):

    def decoder_logsigma(self, sdh):

        if self.enc_dec == 'CNN':

            x = self.deconv_fc_logvar(sdh)

            x = x.view(-1, self.cd[0], self.cd[1], self.cd[2])
@@ -341,6 +341,6 @@ class SDFVAE(nn.Module): 
        ds = torch.cat((d, s_expand), dim=2)

        dsh = torch.cat((ds, h), dim=2)

        recon_x_mu = self.decoder_mu(dsh)

        recon_x_logvar = self.decoder_logvar(dsh)

        return s_mean, s_logvar, s, d_mean, d_logvar, d, d_mean_prior, d_logvar_prior, recon_x_mu, recon_x_logvar

        recon_x_logsigma = self.decoder_logsigma(dsh)

        return s_mean, s_logvar, s, d_mean, d_logvar, d, d_mean_prior, d_logvar_prior, recon_x_mu, recon_x_logsigma