From a73ff8c9125d8286328b025b5870c6e52b822f76 Mon Sep 17 00:00:00 2001 From: sahilg06 Date: Sun, 6 Dec 2020 12:31:24 +0530 Subject: [PATCH] changed the uninitialized variable in multidiscriminator of bicycleGAN --- implementations/bicyclegan/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/implementations/bicyclegan/models.py b/implementations/bicyclegan/models.py index a5701321..91f7f947 100644 --- a/implementations/bicyclegan/models.py +++ b/implementations/bicyclegan/models.py @@ -150,7 +150,7 @@ def discriminator_block(in_filters, out_filters, normalize=True): ), ) - self.downsample = nn.AvgPool2d(in_channels, stride=2, padding=[1, 1], count_include_pad=False) + self.downsample = nn.AvgPool2d(channels, stride=2, padding=[1, 1], count_include_pad=False) def compute_loss(self, x, gt): """Computes the MSE between model output and scalar gt"""