elif v.lower() in ("no", "false", "f", "n", "0"): return False else: raise argparse.ArgumentTypeError("Boolean value expected.") # ---------------------------------------------------------------------------- _examples = ("""examples: python %(prog)s --data-dir=~/datasets --network=gdrive:networks/stylegan2-ffhq-config-f.pkl --metrics=fid50k, ppl_wend --dataset=ffhq --mirror-augment=true valid metrics: """ + ", ".join(sorted([x for x in metric_defaults.keys()])) + """ """) def main(): parser = argparse.ArgumentParser( description="Run StyleGAN2 metrics.", epilog=_examples, formatter_class=argparse.RawDescriptionHelpFormatter, ) parser.add_argument( "--result-dir", help="Root directory for run results (default: %(default)s)", default="results", metavar="DIR", )
#---------------------------------------------------------------------------- _examples = '''examples: # Train StyleGAN2 using the FFHQ dataset python %(prog)s --num-gpus=8 --data-dir=~/datasets --config=config-f --dataset=ffhq --mirror-augment=true valid configs: ''' + ', '.join(_valid_configs) + ''' valid metrics: ''' + ', '.join(sorted([x for x in metric_defaults.keys()])) + ''' ''' def main(): parser = argparse.ArgumentParser( description='Train StyleGAN2.', epilog=_examples, formatter_class=argparse.RawDescriptionHelpFormatter) parser.add_argument( '--result-dir', help='Root directory for run results (default: %(default)s)', default='results', metavar='DIR') parser.add_argument('--data-dir',