From 235fb98837c761f0e752060e2bb15f1ef90c0049 Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Wed, 12 Oct 2022 16:11:21 +0200
Subject: [PATCH] change cgan discriminator back to default
---
mu_map/training/cgan.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/mu_map/training/cgan.py b/mu_map/training/cgan.py
index 0b34ff3..21de59f 100644
--- a/mu_map/training/cgan.py
+++ b/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)
--
GitLab