Add missing semi-colon in discriminator forward sigmoid array range. …

…Fixes a warning about shape (UserWarning: Using a target size (torch.Size([64, 1])) that is different to the input size (torch.Size([64])) is deprecated. Please ensure they have the same size.)

infoGAN.py

@@ -75,7 +75,7 @@ def forward(self, input):
         x = self.conv(input)
         x = x.view(-1, 128 * (self.input_height // 4) * (self.input_width // 4))
         x = self.fc(x)
-        a = F.sigmoid(x[:, self.output_dim])
+        a = F.sigmoid(x[:, :self.output_dim])
         b = x[:, self.output_dim:self.output_dim + self.len_continuous_code]
         c = x[:, self.output_dim + self.len_continuous_code:]
 
@@ -333,4 +333,4 @@ def loss_plot(self, hist, path='Train_hist.png', model_name=''):
 
         path = os.path.join(path, model_name + '_loss.png')
 
-        plt.savefig(path)
+        plt.savefig(path)