You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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
The text was updated successfully, but these errors were encountered:
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:
The text was updated successfully, but these errors were encountered: