Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::TokenEmbeddingConfig Class Referenceexport

Configuration for the TokenEmbedding component. More...

Inheritance diagram for Mila::Dnn::TokenEmbeddingConfig:
Mila::Dnn::ComponentConfig

Public Member Functions

void fromMetadata (const SerializationMetadata &meta) override
 Populate configuration from provided metadata.
dim_t getEmbeddingDim () const
float getEmbeddingScale () const noexcept
dim_t getVocabSize () const
SerializationMetadata toMetadata () const override
 Convert configuration into a SerializationMetadata object.
std::string toString () const override
 Produce a short, human-readable summary of the configuration.
void validate () const override
 Validate configuration parameters.
template<typename Self>
decltype(auto) withEmbeddingDim (this Self &&self, dim_t embedding_dim)
template<typename Self>
decltype(auto) withEmbeddingScale (this Self &&self, float embedding_scale)
template<typename Self>
decltype(auto) withVocabSize (this Self &&self, dim_t vocab_size)
Public Member Functions inherited from Mila::Dnn::ComponentConfig
virtual ~ComponentConfig ()=default
 Virtual destructor for polymorphic base.

Detailed Description

Configuration for the TokenEmbedding component.

Provides a type-safe fluent interface for configuring a pure token embedding lookup. Positional fields are intentionally absent – they belong to the model or attention configuration.

Member Function Documentation

◆ fromMetadata()

void Mila::Dnn::TokenEmbeddingConfig::fromMetadata ( const SerializationMetadata & meta)
inlineoverridevirtual

Populate configuration from provided metadata.

Implementations should read available keys and leave missing keys at their current/default values to preserve forward/backward compatibility.

Parameters
metaMetadata to read configuration values from.

Implements Mila::Dnn::ComponentConfig.

◆ toMetadata()

SerializationMetadata Mila::Dnn::TokenEmbeddingConfig::toMetadata ( ) const
inlineoverridevirtual

Convert configuration into a SerializationMetadata object.

Implementations should include any fields required to fully reconstruct the configuration via fromMetadata.

Returns
SerializationMetadata Metadata representation of the config.

Implements Mila::Dnn::ComponentConfig.

◆ toString()

std::string Mila::Dnn::TokenEmbeddingConfig::toString ( ) const
inlineoverridevirtual

Produce a short, human-readable summary of the configuration.

Implementations should return a compact, single-line description suitable for logging and debugging.

Returns
std::string Human-readable summary of the configuration.

Implements Mila::Dnn::ComponentConfig.

◆ validate()

void Mila::Dnn::TokenEmbeddingConfig::validate ( ) const
inlineoverridevirtual

Validate configuration parameters.

Called by callers to ensure the configuration represents a valid, constructible component. Implementations must throw std::invalid_argument (or a derived exception) when validation fails.

Exceptions
std::invalid_argumentIf the configuration is invalid.

Implements Mila::Dnn::ComponentConfig.


The documentation for this class was generated from the following file: