feat: add option of unet_128_d in --D_netDs

jolibrain · Sep 28, 2023 · 4ad56fd · 4ad56fd
1 parent 05c97f8
commit 4ad56fd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/options/base_options.py b/options/base_options.py
@@ -392,6 +392,7 @@ def initialize(self, parser):
                 "depth",
                 "mask",
                 "sam",
+                "unet_128_d",
             ]
             + list(TORCH_MODEL_CLASSES.keys()),
             help="specify discriminator architecture, another option, --D_n_layers allows you to specify the layers in the n_layers discriminator. NB: duplicated arguments are ignored. Values: basic, n_layers, pixel, projected_d, temporal, vision_aided, depth, mask, sam",