forked from mrq/DL-Art-School
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,
|
||||
logdet=logdet)
|
||||
|
||||
return x, logdet
|
||||
return x, logdet, lr_enc['out']
|
||||
|
|
Loading…
Reference in New Issue
Block a user