ML_modules.build_mlp#
Back to Module page
Summary#
Create an multi-layer perceptron (MLP) from the configuration.
Signature#
def build_mlp(input_dim, hidden_dim, output_dim, layers, activation = nn.GELU())
Name |
Type |
Default |
Description |
|---|---|---|---|
|
|||
|
|||
|
|||
|
|||
|
nn.GELU() |
Output variables#
Return annotation |
Docstring type |
Description |
|---|---|---|
|
Docstring#
Create an multi-layer perceptron (MLP) from the configuration.