[architectures] DR-GAN discriminator: added the number of ids as class...

[architectures] DR-GAN discriminator: added the number of ids as class variables, fixed the size of the output

bob/learn/pytorch/architectures/DRGAN.py

@@ -163,7 +163,8 @@ class DRGAN_discriminator(nn.Module):
   def __init__(self, image_size, number_of_ids, conditional_dim):
     
     super(DRGAN_discriminator, self).__init__()
-    
+   
+    self.number_of_ids = number_of_ids
     self.ngpu = 1
     n_fm_first = 64
        
@@ -202,4 +203,4 @@ class DRGAN_discriminator(nn.Module):
     else:
       output = self.main(x)
 
-    return output
+    return output.squeeze()