unilab.algos.mlx.common.mlp.MLP
-
class unilab.algos.mlx.common.mlp.MLP[source]
Bases: Module
Simple feed-forward MLP with configurable activations.
- Parameters:
-
Methods
__init__(input_dim, output_dim, hidden_dims)
|
|
init_orthogonal([hidden_gain, output_gain])
|
Orthogonally initialize linear layers with separate output gain. |
-
__init__(input_dim, output_dim, hidden_dims, activation='elu', last_activation=None)[source]
- Parameters:
-
-
__call__(x)[source]
Call self as a function.
- Parameters:
x (array)
- Return type:
array
-
init_orthogonal(hidden_gain=1.4142135623730951, output_gain=1.0)[source]
Orthogonally initialize linear layers with separate output gain.
- Parameters:
-
- Return type:
None