save_samples.py 文件源码

def main():
    artist = 'kanye_west'
    model_path = '../save/models/kanye_west/kanye_west.ckpt-30000'
    num_save = 1000

    try:
        opts, _ = getopt.getopt(sys.argv[1:], 'l:a:N:', ['load_path=', 'artist_name=', 'num_save='])
    except getopt.GetoptError:
        sys.exit(2)

    for opt, arg in opts:
        if opt in ('-l', '--load_path'):
            model_path = arg
        if opt in ('-a', '--artist_name'):
            artist = arg
        if opt in ('-n', '--num_save'):
            num_save = int(arg)

    save(artist, model_path, num_save)