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

Revert gaussian loss changes

Browse Source
This commit is contained in:
James Betker 2020-12-04 12:49:20 -07:00
parent 7da2a0d5d4
commit 7a81d4e2f4
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -127,11 +127,7 @@ class SRFlowNet(nn.Module):
else:
z = epses
logp = 0
for eps in epses:
logp = logp + flow.GaussianDiag.logp(None, None, eps)
logp_weight = opt_get(self.opt, ['networks', 'generator', 'flow', 'gaussian_loss_weight'], 1)
logp = logp * logp_weight
logp = flow.GaussianDiag.logp(None, None, z)
objective = objective + logp
nll = (-objective) / float(np.log(2.) * pixels)