-
Notifications
You must be signed in to change notification settings - Fork 27.5k
Improve save() method in torch.jit.ScriptModule #17688
Copy link
Copy link
Open
Labels
jit-backloglow priorityWe're unlikely to get around to doing this in the near futureWe're unlikely to get around to doing this in the near futureoncall: jitAdd this issue/PR to JIT oncall triage queueAdd this issue/PR to JIT oncall triage queuetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
Metadata
Metadata
Assignees
Labels
jit-backloglow priorityWe're unlikely to get around to doing this in the near futureWe're unlikely to get around to doing this in the near futureoncall: jitAdd this issue/PR to JIT oncall triage queueAdd this issue/PR to JIT oncall triage queuetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
🚀 Feature
save() method in torch.jit.ScriptModule should support save_to_buffer as well.
Motivation
Right now, to save a torch.jit.ScriptModule to buffer, I have to call function torch.jit.save().
It would be more convenient and consistent if torch.jit.ScriptModule.save() can perform the same job.
Pitch
I can work on this feature.
cc @suo