SRFlow outputs RRDB output
This commit is contained in:
parent
c25b49bb12
commit
49327b99fe
|
@ -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']
|
||||||
|
|
Loading…
Reference in New Issue
Block a user