def parse_unstructured_arguments()

in args.py [0:0]

• 34 lines of code
• 2 McCabe index (conditional complexity)

def parse_unstructured_arguments():
    parser = gen_args("Unstructured Sparsity Training")

    parser.add_argument(
        "--method",
        type=str,
        default="lcs_l",
        help="Training method. One of topk_target, lcs_p, lcs_l",
    )

    parser.add_argument(
        "--topk",
        type=float,
        help="Target topk (0,1) for BN training",
    )

    parser.add_argument(
        "--topk_lower_bound",
        type=float,
        help="Lower bound (high accuracy) endpoint of the learned subspace",
    )

    parser.add_argument(
        "--topk_upper_bound",
        type=float,
        help="Upper bound (high efficiency) endpoint of the learned subspace",
    )

    parser.add_argument(
        "--warmup_budget",
        type=float,
        help="Value in range (0,100] denoting the percentage of epochs for warmup phase",
    )

    parser.add_argument(
        "--eval_topk_grid",
        type=lambda x: [float(w) for w in x.split(",")],
        help="Will evaluate at these topk values",
    )

    return parser.parse_args()