change cgan discriminator back to default

mu_map/training/cgan.py

@@ -421,8 +421,8 @@ if __name__ == "__main__":
     torch.manual_seed(args.seed)
     np.random.seed(args.seed)
 
-    # discriminator = Discriminator(in_channels=2, input_size=args.patch_size)
-    discriminator = PatchDiscriminator(in_channels=2, input_size=args.patch_size)
+    discriminator = Discriminator(in_channels=2, input_size=args.patch_size)
+    # discriminator = PatchDiscriminator(in_channels=2, input_size=args.patch_size)
     discriminator = discriminator.to(device)
 
     generator = UNet(in_channels=1, features=args.features)