def get_parser()

in sing/generate.py [0:0]

• 40 lines of code
• 1 McCabe index (conditional complexity)

def get_parser():
    parser = argparse.ArgumentParser(
        "sing.generate",
        description="Generate audio samples from a trained SING model",
        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument(
        "--model",
        type=Path,
        default="models/sing.th",
        help="Path to the trained SING model as outputted by sing.main")
    parser.add_argument(
        "--dl",
        action="store_true",
        help="Download if necessary a pretrained SING model.")
    parser.add_argument(
        "--output",
        type=Path,
        default="generated",
        help="Path where the generated samples will be saved")
    parser.add_argument(
        "--metadata",
        default=nsynth.get_metadata_path(),
        type=Path,
        help="path to the dataset metadata file")

    parser.add_argument(
        "list",
        type=Path,
        help="File containing a list of names from the nsynth dataset. "
        "Those notes will be generated by SING")
    parser.add_argument(
        "--batch-size", type=int, default=32, help="Batch size")
    parser.add_argument("--cuda", action="store_true", help="Use cuda")
    parser.add_argument(
        "--parallel", action="store_true", help="Use multiple gpus")
    parser.add_argument(
        "--unpad",
        default=512,
        type=int,
        help="Amount of unpadding to perform")
    return parser