SRFlow outputs RRDB output

2020-12-16 10:28:02 -07:00 · 2020-12-16 10:28:02 -07:00 · 49327b99fe
commit 49327b99fe
parent c25b49bb12
1 changed files with 1 additions and 1 deletions
--- a/codes/models/archs/srflow_orig/SRFlowNet_arch.py
+++ b/codes/models/archs/srflow_orig/SRFlowNet_arch.py
@ -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']