From 6f958bb15004f5cac9514c13f640bc96678d27b7 Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 27 Nov 2020 15:21:13 -0700 Subject: [PATCH] Maybe this is necessary after all? --- codes/models/archs/srflow_orig/Split.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/codes/models/archs/srflow_orig/Split.py b/codes/models/archs/srflow_orig/Split.py index f6d0aff7..5aeb2eeb 100644 --- a/codes/models/archs/srflow_orig/Split.py +++ b/codes/models/archs/srflow_orig/Split.py @@ -37,8 +37,7 @@ class Split2d(nn.Module): eps = (z2 - mean) / self.exp_eps(logs) - # This has been moved into SRFlowNet_arch.py alongside the other Z NLL losses. - # logdet = logdet + self.get_logdet(logs, mean, z2) + logdet = logdet + self.get_logdet(logs, mean, z2) # print(logs.shape, mean.shape, z2.shape) # self.eps = eps @@ -57,8 +56,7 @@ class Split2d(nn.Module): z2 = mean + self.exp_eps(logs) * eps z = thops.cat_feature(z1, z2) - # This has been moved into SRFlowNet_arch.py alongside the other Z NLL losses. - #logdet = logdet - self.get_logdet(logs, mean, z2) + logdet = logdet - self.get_logdet(logs, mean, z2) return z, logdet # return z, logdet, eps