dataeval.utils.onnx¶
Utilities for working with ONNX models.
These utilities help prepare ONNX models for embedding extraction by identifying and exposing intermediate layers that produce embeddings.
Functions¶
Find the embedding layer name in an ONNX classification model. |
|
Modify an ONNX model to output embeddings from an intermediate layer. |