From dccd8e53c668a532792d46302caec7419de6e2be Mon Sep 17 00:00:00 2001 From: sniklaus Date: Fri, 24 Nov 2017 14:51:29 -0800 Subject: no message --- run.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) (limited to 'run.py') diff --git a/run.py b/run.py index a1f26a3..a7a7add 100644 --- a/run.py +++ b/run.py @@ -142,13 +142,19 @@ class Network(torch.nn.Module): variableDeconv5 = self.moduleDeconv5(variablePool5) variableUpsample5 = self.moduleUpsample5(variableDeconv5) - variableDeconv4 = self.moduleDeconv4(variableUpsample5 + variableConv5) + variableCombine = variableUpsample5 + variableConv5 + + variableDeconv4 = self.moduleDeconv4(variableCombine) variableUpsample4 = self.moduleUpsample4(variableDeconv4) - variableDeconv3 = self.moduleDeconv3(variableUpsample4 + variableConv4) + variableCombine = variableUpsample4 + variableConv4 + + variableDeconv3 = self.moduleDeconv3(variableCombine) variableUpsample3 = self.moduleUpsample3(variableDeconv3) - variableDeconv2 = self.moduleDeconv2(variableUpsample3 + variableConv3) + variableCombine = variableUpsample3 + variableConv3 + + variableDeconv2 = self.moduleDeconv2(variableCombine) variableUpsample2 = self.moduleUpsample2(variableDeconv2) variableCombine = variableUpsample2 + variableConv2 @@ -181,8 +187,8 @@ intPaddingLeft = int(math.floor(51 / 2.0)) intPaddingTop = int(math.floor(51 / 2.0)) intPaddingRight = int(math.floor(51 / 2.0)) intPaddingBottom = int(math.floor(51 / 2.0)) -modulePaddingInput = torch.nn.Module() -modulePaddingOutput = torch.nn.Module() +modulePaddingInput = torch.nn.Sequential() +modulePaddingOutput = torch.nn.Sequential() if True: intPaddingWidth = intPaddingLeft + intWidth + intPaddingRight -- cgit v1.2.3-70-g09d2