Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

RTen models should have provenance metadata #20

Closed
robertknight opened this issue Dec 31, 2023 · 1 comment
Closed

RTen models should have provenance metadata #20

robertknight opened this issue Dec 31, 2023 · 1 comment
Labels
tooling Tools for debugging / profiling etc.

Comments

@robertknight
Copy link
Owner

robertknight commented Dec 31, 2023

RTen models should have machine-readable metadata in them that specifies where they came from. At a minimum this would include:

  • Filename and hash (SHA-1 ?) of the ONNX model that was used to generate it
  • Date when the RTen model was created

This enables verifying that a .rten model corresponds to a particular ONNX model. The rten-cli package could support a --metadata flag that would dump this info from a compiled model.

In addition it would be useful to also incorporate metadata from the ONNX model. See https://github.com/onnx/onnx/blob/main/docs/IR.md#models. In particular I would like to encourage including:

  • The URL of the associated repositories
  • The license
  • A description of the dataset used to train the model
  • Version of the dataset and or tools used to train the model
@robertknight
Copy link
Owner Author

Basic support for this was added in #48. I will flesh it out with more fields as needed.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
tooling Tools for debugging / profiling etc.
Projects
None yet
Development

No branches or pull requests

1 participant