SRFlow outputs RRDB output

This commit is contained in:
James Betker 2020-12-16 10:28:02 -07:00
parent c25b49bb12
commit 49327b99fe

View File

@ -166,4 +166,4 @@ class SRFlowNet(nn.Module):
x, logdet = self.flowUpsamplerNet(rrdbResults=lr_enc, z=z, eps_std=eps_std, reverse=True, epses=epses, x, logdet = self.flowUpsamplerNet(rrdbResults=lr_enc, z=z, eps_std=eps_std, reverse=True, epses=epses,
logdet=logdet) logdet=logdet)
return x, logdet return x, logdet, lr_enc['out']