.. Custom template for autosummary classes. Can be removed when
.. https://github.com/sphinx-doc/sphinx/issues/7912 is fixed.

dgs.models.embedding\_generator.torchreid.TorchreidEmbeddingGenerator
=====================================================================

.. currentmodule:: dgs.models.embedding_generator.torchreid

.. autoclass:: TorchreidEmbeddingGenerator

   
   

   .. rubric:: Methods

   

   .. automethod:: TorchreidEmbeddingGenerator.__init__

   .. automethod:: TorchreidEmbeddingGenerator.configure_torch_module

   .. automethod:: TorchreidEmbeddingGenerator.embedding_key_exists

   .. automethod:: TorchreidEmbeddingGenerator.forward

   .. automethod:: TorchreidEmbeddingGenerator.predict_embeddings

   .. automethod:: TorchreidEmbeddingGenerator.predict_ids

   .. automethod:: TorchreidEmbeddingGenerator.terminate

   .. automethod:: TorchreidEmbeddingGenerator.validate_params
   

   

   
       
           .. rubric:: Attributes

           .. autosummary::
              :template: attribute.rst
              :toctree:
           
              ~TorchreidEmbeddingGenerator.device
              ~TorchreidEmbeddingGenerator.is_training
              ~TorchreidEmbeddingGenerator.module_name
              ~TorchreidEmbeddingGenerator.module_type
              ~TorchreidEmbeddingGenerator.name
              ~TorchreidEmbeddingGenerator.name_safe
              ~TorchreidEmbeddingGenerator.precision
              ~TorchreidEmbeddingGenerator.model
              ~TorchreidEmbeddingGenerator.embedding_size
              ~TorchreidEmbeddingGenerator.nof_classes