Skip to content

Support ingesting PyTorch models #10

@pcolange

Description

@pcolange

This Issue is a feature request for the ability to load PyTorch models from a *.pth file into the GUI. Natively supporting PyTorch would require a new model summary GUI page, so we opt for this first pass to export PyTorch models as ONNX models so we can leverage all the current capabilities without requiring new PyTorch capabilities.

The story for this feature will include these steps:

  1. Enable *.pth files to be opened into Digest AI
  2. Open a new ingest window to enable users to set input dtypes and shapes and configure other export options
  3. Export the PyTorch model as an ONNX model
  4. Load exported ONNX model into Digest using the already supported ONNX model path

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions