User Guide:
User API:
Developer API:
Full API:
Bases: object
object
Embedding extractor from a layer :param model: Model used for inference. :type model: nn.Module
Methods
disp_modules
Display the the modules of the model.
register
Registration for embedding extraction.
Registration for embedding extraction. :param layer_name: Name of the layer from which the embedding is extracted. :type layer_name: str