diff --git a/ginka/save.py b/ginka/save.py new file mode 100644 index 0000000..e671f2f --- /dev/null +++ b/ginka/save.py @@ -0,0 +1,16 @@ +import argparse +import torch + +def to_deployment(path: str, output: str): + state = torch.load(path) + torch.save({ + "model_state": state["model_state"] + }, output) + +if __name__ == '__main__': + parser = argparse.ArgumentParser() + parser.add_argument("--input", type=str, default="result/ginka.pth") + parser.add_argument("--output", type=str, default="result/ginka_deploy.pth") + args = parser.parse_args() + to_deployment(args.input, args.output) + \ No newline at end of file