From 6c8c35ac47d9bc13d65177db31db075427e40c7e Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 20 Nov 2020 10:03:06 -0700 Subject: [PATCH] Support training RRDB encoder [srflow] --- codes/models/archs/srflow_orig/SRFlowNet_arch.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/codes/models/archs/srflow_orig/SRFlowNet_arch.py b/codes/models/archs/srflow_orig/SRFlowNet_arch.py index 2bdab09d..b0e909f0 100644 --- a/codes/models/archs/srflow_orig/SRFlowNet_arch.py +++ b/codes/models/archs/srflow_orig/SRFlowNet_arch.py @@ -25,10 +25,7 @@ class SRFlowNet(nn.Module): hidden_channels = opt_get(opt, ['networks', 'generator','flow', 'hidden_channels']) hidden_channels = hidden_channels or 64 - self.RRDB_training = True # Default is true - - train_RRDB_delay = opt_get(self.opt, ['networks', 'generator','train_RRDB_delay']) - self.RRDB_training = False + self.RRDB_training = opt_get(self.opt, ['networks', 'generator','train_RRDB'], default=False) self.flowUpsamplerNet = \ FlowUpsamplerNet((160, 160, 3), hidden_channels, K,