Language Language English 简体中文

unilab.algos.mlx.common.mlp

MLP module used by MLX RL algorithms.

Classes

MLP	Simple feed-forward MLP with configurable activations.

class unilab.algos.mlx.common.mlp.MLP

Bases: Module

Simple feed-forward MLP with configurable activations.

Parameters:

• input_dim (int)

• output_dim (int)

• hidden_dims (Sequence[int])

• activation (str)

• last_activation (str | None)

__init__(input_dim, output_dim, hidden_dims, activation='elu', last_activation=None)

Parameters:

• input_dim (int)

• output_dim (int)

• hidden_dims (Sequence[int])

• activation (str)

• last_activation (str | None)

init_orthogonal(hidden_gain=1.4142135623730951, output_gain=1.0)

Orthogonally initialize linear layers with separate output gain.

Parameters:

• hidden_gain (float)

• output_gain (float)

Return type:

None