Update model/models.py

model/models.py

@@ -186,7 +186,7 @@ class Generator(nn.Module):
         
         self.feature_conv = FeatureConv()
 
-        self.to0 = self._make_encoder_block_first(input_channels, 64)
+        self.to0 = self._make_encoder_block_first(input_channels, 32)
         self.to1 = self._make_encoder_block(32, 64)
         self.to2 = self._make_encoder_block(64, 128)
         self.to3 = self._make_encoder_block(128, 256)