-
Notifications
You must be signed in to change notification settings - Fork 2
Open
Labels
enhancementNew feature or requestNew feature or request
Description
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:
- Enable *.pth files to be opened into Digest AI
- Open a new ingest window to enable users to set input dtypes and shapes and configure other export options
- Export the PyTorch model as an ONNX model
- Load exported ONNX model into Digest using the already supported ONNX model path
mshr-h
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request