Fix ref branch using fixed filters

2020-09-11 08:58:35 -06:00 · 2020-09-11 08:58:35 -06:00 · 1086f0476b
commit 1086f0476b
parent 8c469b8286
1 changed files with 2 additions and 2 deletions
--- 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):