Update model/models.py
Browse files- model/models.py +1 -1
model/models.py
CHANGED
@@ -186,7 +186,7 @@ class Generator(nn.Module):
|
|
186 |
|
187 |
self.feature_conv = FeatureConv()
|
188 |
|
189 |
-
self.to0 = self._make_encoder_block_first(input_channels,
|
190 |
self.to1 = self._make_encoder_block(32, 64)
|
191 |
self.to2 = self._make_encoder_block(64, 128)
|
192 |
self.to3 = self._make_encoder_block(128, 256)
|
|
|
186 |
|
187 |
self.feature_conv = FeatureConv()
|
188 |
|
189 |
+
self.to0 = self._make_encoder_block_first(input_channels, 32)
|
190 |
self.to1 = self._make_encoder_block(32, 64)
|
191 |
self.to2 = self._make_encoder_block(64, 128)
|
192 |
self.to3 = self._make_encoder_block(128, 256)
|