fix a bug in decoders

Author: garjania

models/distributed_model.py

@@ -107,7 +107,7 @@ def decode(self, x, w):
         x = self.igdn1(x)
         x = self.deconv2(x)
         x = self.igdn2(x)
-        x = self.deconv2(x)
+        x = self.deconv3(x)
         x = self.igdn3(x)
         x = self.deconv4(x)
 
@@ -119,7 +119,7 @@ def decode_cor(self, x, w):
         x = self.igdn1_cor(x)
         x = self.deconv2_cor(x)
         x = self.igdn2_cor(x)
-        x = self.deconv2_cor(x)
+        x = self.deconv3_cor(x)
         x = self.igdn3_cor(x)
         x = self.deconv4_cor(x)
 
@@ -245,7 +245,7 @@ def decode(self, x, w):
         x = self.igdn1(x)
         x = self.deconv2(x)
         x = self.igdn2(x)
-        x = self.deconv2(x)
+        x = self.deconv3(x)
         x = self.igdn3(x)
         x = self.deconv4(x)
         return x
@@ -256,7 +256,7 @@ def decode_cor(self, x, w):
         x = self.igdn1_cor(x)
         x = self.deconv2_cor(x)
         x = self.igdn2_cor(x)
-        x = self.deconv2_cor(x)
+        x = self.deconv3_cor(x)
         x = self.igdn3_cor(x)
         x = self.deconv4_cor(x)
         return x