From af2a39fd5dec236cf8a169f6956de25be3f7d3e8 Mon Sep 17 00:00:00 2001 From: John Pope Date: Mon, 12 Aug 2024 11:07:20 +1000 Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 6ef4a61..70595e0 100644 --- a/train.py +++ b/train.py @@ -419,7 +419,7 @@ def main(): d0 = IMFPatchDiscriminator(ndf=config.discriminator.ndf) d1 = MultiScalePatchDiscriminator(input_nc=3, ndf=64, n_layers=3, num_D=3) - discriminator = d1 if config.training.use_multiscale_discriminator else do + discriminator = d1 if config.training.use_multiscale_discriminator else d0 add_gradient_hooks(discriminator)