tlc.integration.super_gradients.callbacks.embeddings_pipeline¶
SuperGradients pipeline for computing embedding for Yolo-NAS models.
Module Contents¶
Classes¶
Class |
Description |
|---|---|
API¶
- class EmbeddingsPipeline(
- model: Module,
- image_processor: super_gradients.training.processing.processing.Processing | list[super_gradients.training.processing.processing.Processing] = None,
- device: str | None = None,
- fp16: bool = True,
- use_global_pooling: bool = True,
Bases:
super_gradients.training.pipelines.pipelines.Pipeline