1b50e0ce69cff758a6ed4cef847c01e9aa5b3f2d,tools/release_model.py,,,#,5

Before Change

import torch

if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description="Removes the optim data of PyTorch models")
    parser.add_argument("--model", "-m",
                        help="The model filename (*.pt)", required=True)
    parser.add_argument("--output", "-o",
                        help="The output filename (*.pt)", required=True)
    opt = parser.parse_args()

    model = torch.load(opt.model)
    model["optim"] = None
    torch.save(model, opt.output)

After Change

from onmt.bin.release_model import main


if __name__ == "__main__":
    main()

Instances