-
Notifications
You must be signed in to change notification settings - Fork 6
Description
🚀 Feature Request
Chaining of the export and package steps provided in PR #97 might be very commonly performed. This chain of operations should be directly within the stac_model.torch utilities to allow users directly working with stac_model or MLM definitions to package their models as .pt2 archives more efficiently without dealing with intermediate files.
Potentially, there could be a stac_model.torch.save function that takes various combinations of nn.Module | ExportedProgram and deals with the branching/sub-tooling calls automatically to save the result.
Also considering crim-ca/mlm-extension#41 (comment), there could be a format option to that stac_model.torch.save function that can deal with STAC Item JSON vs MLM-metadata YAML representation.
🔉 Motivation
Avoid users having to understand all the intricacies of packaging a model if they can provide minimally viable definitions via MLM metadata and the nn.Module reference.
📡 Alternatives
Refer to the unit test example doing the same chain of operations in PR #97.
📎 Additional context
Originally posted by @fmigneault in #97 (comment)