🧗

johndpope · Aug 12, 2024 · af2a39f · af2a39f
1 parent 9761e70
commit af2a39f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)