Forgot to add new configs to the parser

PDillis · Apr 28, 2021 · 08378e1 · 08378e1
1 parent 1db9108
commit 08378e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -448,7 +448,7 @@ def convert(self, value, param, ctx):
 @click.option('--mirror-y', help='Enable dataset y=flips [default: false]', type=bool, metavar='BOOL')
 
 # Base config.
-@click.option('--cfg', help='Base config [default: auto]', type=click.Choice(['auto', 'stylegan2', 'paper256', 'paper512', 'paper1024', 'cifar', '24gb-2gpu']))
+@click.option('--cfg', help='Base config [default: auto]', type=click.Choice(['auto', 'stylegan2', 'paper256', 'paper512', 'paper1024', 'cifar', '24gb-2gpu', '24gb-4gpu', '24gb-2gpu-cplx', '24gb-4gpu-cplx']))
 @click.option('--gamma', help='Override R1 gamma', type=float)
 @click.option('--kimg', help='Override training duration', type=int, metavar='INT')
 @click.option('--batch', help='Override batch size', type=int, metavar='INT')