MLPEmbedding.Config

Component: MLPEmbedding

class MLPEmbedding.Config

Bases: Module.Config

All Attributes (including base classes)

load_path: Optional[str] = None

save_path: Optional[str] = None

freeze: bool = False

shared_module_key: Optional[str] = None

embed_dim: int = 100

embedding_init_strategy: EmbedInitStrategy = <EmbedInitStrategy.RANDOM: 'random'>

embedding_init_range: Optional[list[float]] = None

embeddding_init_std: Optional[float] = 0.02

export_input_names: list[str] = ['mlp_vals']

mlp_layer_dims: list[int] = []

cpu_only: bool = False

skip_header: bool = True

delimiter: str = ' '

Default JSON

{
    "load_path": null,
    "save_path": null,
    "freeze": false,
    "shared_module_key": null,
    "embed_dim": 100,
    "embedding_init_strategy": "random",
    "embedding_init_range": null,
    "embeddding_init_std": 0.02,
    "export_input_names": [
        "mlp_vals"
    ],
    "mlp_layer_dims": [],
    "cpu_only": false,
    "skip_header": true,
    "delimiter": " "
}