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',
