cmd.py

import os

cmd_type = 'synthesize'
assert cmd_type in ['synthesize', 'build', 'train']

generator_name = 'arc_pavement'

if cmd_type == 'synthesize':
    cmd = 'python synthesis.py ' \
          '--data_path=./data/sbs ' \
          f'--generator_name={generator_name} ' \
          '--n_samples=512 ' \
          '--verify '

elif cmd_type == 'build':
    cmd = f'python stylegan/dataset_tool.py --source=./data/sbs/{generator_name} ' \
          f'--dest=./data/train/{generator_name}_300k.zip'

elif cmd_type == 'train':
    cmd = 'python stylegan/train.py ' \
        '--cfg stylegan2 ' \
        '--outdir=./training-runs ' \
        f'--data=./data/train/{generator_name}_300k.zip ' \
        f'--generator_name={generator_name} ' \
        '--gpus=1 ' \
        '--batch=8 ' \
        '--gamma=10 ' \
        '--map-depth=4 ' \
        '--glr=0.0025 ' \
        '--dlr=0.001 ' \
        '--cbase=16384 ' \
        '--snap=50 ' \
        '--cond=true ' \
        '--aug=noaug ' \
        '--kimg=10000 ' \
        '--metrics=none ' \
        '--mirror=false ' \
        '--norm_type=norm ' \
        '--no_gan=True ' \
        '--cond_d=True ' \
        # '--dry-run'
else:
    raise NotImplementedError

print('*' * 50)
print(cmd)
os.system(cmd)