From 49327b99feabd2651061fcce46807fa9b5de07c4 Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 16 Dec 2020 10:28:02 -0700 Subject: [PATCH] SRFlow outputs RRDB output --- codes/models/archs/srflow_orig/SRFlowNet_arch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/archs/srflow_orig/SRFlowNet_arch.py b/codes/models/archs/srflow_orig/SRFlowNet_arch.py index b5959f2b..e468b50e 100644 --- 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']