Merge branch 'main' of https://github.com/johndpope/IMF

johndpope · Aug 28, 2024 · 6def659 · 6def659
2 parents 42b8bd5 + f9725b5
commit 6def659
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -295,8 +295,8 @@ def main():
         use_resnet_feature=config.model.use_resnet_feature
     )
 
-    discriminator = IMFPatchDiscriminator()
-
+    # discriminator = IMFPatchDiscriminator()
+    discriminator = MultiScalePatchDiscriminator(input_nc=3, ndf=64, n_layers=3, num_D=3)
     # transform = transforms.Compose([
     #     transforms.Resize((256, 256)),
     #     transforms.ToTensor(),