mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Maybe this is necessary after all?

Browse Source
This commit is contained in:
James Betker 2020-11-27 15:21:13 -07:00
parent ef8d5f88c1
commit 6f958bb150
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
codes/models/archs/srflow_orig/Split.py
View File

@ -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