From 1086f0476be8c06c2e9ba3b0414ad92c629f672a Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 11 Sep 2020 08:58:35 -0600 Subject: [PATCH] Fix ref branch using fixed filters --- codes/models/archs/SPSR_arch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/codes/models/archs/SPSR_arch.py b/codes/models/archs/SPSR_arch.py index e22fcc64..c98523f4 100644 --- a/codes/models/archs/SPSR_arch.py +++ b/codes/models/archs/SPSR_arch.py @@ -359,8 +359,8 @@ class SwitchedSpsr(nn.Module): class RefJoiner(nn.Module): def __init__(self, nf): super(RefJoiner, self).__init__() - self.lin1 = nn.Linear(512, 256) - self.lin2 = nn.Linear(256, nf) + self.lin1 = nn.Linear(nf * 8, nf * 4) + self.lin2 = nn.Linear(nf * 4, nf) self.join = ReferenceJoinBlock(nf, residual_weight_init_factor=.1) def forward(self, x, ref):