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. |