e0b692c96565074d4cb148e6011b2007d0a55348,examples/model_compress/model_speedup.py,,,#,129

Before Change

            args.masks_file = "mask_vgg16_cifar10.pth"
        apoz_speedup(args.masks_file, args.model_checkpoint)
    else:
        raise ValueError("unsupported example_name: {}".format(args.example_name))

After Change

    if args.example_name != "all":
        if args.masks_file is not None:
            config[args.example_name]["masks_file"] = args.masks_file
        if not os.path.exists(config[args.example_name]["masks_file"]):
            msg = "{} does not exist! You should specify masks_file correctly, " \
                  "or use default one which is generated by model_prune_torch.py"
            raise RuntimeError(msg.format(config[args.example_name]["masks_file"]))
        model_inference(config[args.example_name])
    else:
        model_inference(config["fpgm"])
        model_inference(config["slim"])

Instances