def create_arg_parser()

in fastmri_examples/cs/run_bart.py [0:0]

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

def create_arg_parser():
    parser = ArgumentParser(add_help=False)

    parser.add_argument(
        "--data_path",
        type=pathlib.Path,
        required=True,
        help="Path to the data",
    )
    parser.add_argument(
        "--output_path",
        type=pathlib.Path,
        required=True,
        help="Path to save the reconstructions to",
    )
    parser.add_argument(
        "--challenge",
        type=str,
        required=True,
        help="Which challenge",
    )
    parser.add_argument(
        "--sample_rate",
        type=float,
        default=1.0,
        help="Percent of data to run",
    )
    parser.add_argument(
        "--mask_type", choices=["random", "equispaced"], default="random", type=str
    )
    parser.add_argument(
        "--split",
        choices=["train", "val", "test", "challenge"],
        default="val",
        type=str,
    )
    parser.add_argument("--accelerations", nargs="+", default=[4], type=int)
    parser.add_argument("--center_fractions", nargs="+", default=[0.08], type=float)

    # bart args
    parser.add_argument(
        "--num_iters",
        type=int,
        default=200,
        help="Number of iterations to run the reconstruction algorithm",
    )
    parser.add_argument(
        "--reg_wt", type=float, default=0.01, help="Regularization weight parameter"
    )
    parser.add_argument(
        "--num_procs",
        type=int,
        default=4,
        help="Number of processes. Set to 0 to disable multiprocessing.",
    )

    return parser