xpag.agents.rljax_agents.network.actor.StateIndependentGaussianPolicy

class StateIndependentGaussianPolicy(action_dim, hidden_units=(64, 64))

Bases: Module

Policy for PPO.

Initializes the current module with the given name.

Subclasses should call this constructor before creating other modules or variables such that those modules are named correctly.

Parameters:

name – An optional string name for the class. Must be a valid Python identifier. If name is not provided then the class name for the current instance is converted to lower_snake_case and used instead.

Methods

params_dict	Returns parameters keyed by name for this module and submodules.
state_dict	Returns state keyed by name for this module and submodules.

__call__(x)

Call self as a function.

params_dict()

Returns parameters keyed by name for this module and submodules.

Return type:

Mapping[str, Array]

state_dict()

Returns state keyed by name for this module and submodules.

Return type:

Mapping[str, Array]