From 902527dfaa623d14263a592b110d8404ae75ed7f Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 10 Jul 2020 23:00:21 -0600 Subject: [PATCH] err4 --- codes/models/archs/arch_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/archs/arch_util.py b/codes/models/archs/arch_util.py index 47a64b2d..c239bc60 100644 --- a/codes/models/archs/arch_util.py +++ b/codes/models/archs/arch_util.py @@ -374,7 +374,7 @@ class ExpansionBlock(nn.Module): filters_out = filters_in // 2 self.decimate = block(filters_in, filters_out, kernel_size=1, bias=False, activation=False, norm=True) self.process_passthrough = block(filters_out, filters_out, kernel_size=3, bias=True, activation=False, norm=True) - self.conjoin = block(filters_in, filters_out, kernel_size=3, bias=False, activation=True, norm=False) + self.conjoin = block(filters_out*2, filters_out, kernel_size=3, bias=False, activation=True, norm=False) self.process = block(filters_out, filters_out, kernel_size=3, bias=False, activation=True, norm=True) # input is the feature signal with shape (b, f, w, h)