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.

Sign up for GitHub

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

Jump to bottom

Add initial support for model metadata in RTen models #48

Merged
merged 4 commits into from
Feb 4, 2024
Merged

Add initial support for model metadata in RTen models #48

merged 4 commits into from
Feb 4, 2024

Commits on Feb 4, 2024

  1. Reformat convert-onnx.py with black 24.1.1

    @robertknight
    robertknight committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    22be907 View commit details
    Browse the repository at this point in the history

  2. Support storing model metadata in RTen models 

    Add a new `Metadata` table to the FlatBuffers schema and `Model::metadata`
structure for accessing this data in Rust.

The initial set of fields includes a description, license and metadata
describing the provenance of the model:

 - The SHA-256 hash of the input ONNX model
 - The code repository (eg. GitHub)
 - The model repository (eg. Hugging Face)
 - Training run ID and URL (eg. from an experiment tracking platform such as
   Weights and Biases)
    @robertknight
    robertknight committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    feb98fb View commit details
    Browse the repository at this point in the history

  3. Display model metadata in rten CLI output

    @robertknight
    robertknight committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    62233e9 View commit details
    Browse the repository at this point in the history

  4. Support adding metadata to RTen models when converting from ONNX 

    This initial support assumes that the metadata is stored separately from the
ONNX file. In future it will also be useful to import from metadata stored
within the ONNX file itself.
    @robertknight
    robertknight committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    b4adf51 View commit details
    Browse the repository at this point in the history