40c5039380d45c96ddbd3fd951e5c9adf167647d,tests/test_modules.py,,build_varnet_args,#Any#,66

Before Change

    num_gpus = 2 if backend == "ddp" else 1
    batch_size = 1

    config = dict(
        num_cascades=8,
        pools=4,
        chans=18,
        sens_pools=4,
        sens_chans=8,
        mask_type="equispaced",
        center_fractions=[0.08],
        accelerations=[4],
        resolution=384,
        lr=0.001,
        lr_step_size=40,
        lr_gamma=0.1,
        weight_decay=0.0,
        data_path=knee_path,
        challenge="multicoil",
        exp_dir=logdir,
        exp_name="varnet_demo",
        test_split="test",
        batch_size=batch_size,
    )
    parser.set_defaults(**config)

    parser.set_defaults(
        gpus=num_gpus,

After Change

        type=float,
        help="Number of center lines to use in mask",
    )
    parser.add_argument(
        "--accelerations",
        nargs="+",
        default=[4],
        type=int,
        help="Acceleration rates to use for masks",
    )

    // data config
    parser = FastMriDataModule.add_data_specific_args(parser)
    parser.set_defaults(

Instances