dgs.models.embedding_generator.embedding_generator.EmbeddingGeneratorModule.forward

abstract EmbeddingGeneratorModule.forward(ds: State) → torch.Tensor

Predict next outputs, given any data in a State object, using this Re-ID model.

Returns:

The generated embeddings as tensor of shape [N x embedding_size].