xpag.agents.rljax_agents.network.critic.ContinuousQuantileFunction#

class ContinuousQuantileFunction(num_critics=5, hidden_units=(512, 512, 512), num_quantiles=25, d2rl=False)#

Bases: Module

Critic for TQC.

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__(s, a)#

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]