ray.rllib.algorithms.algorithm_config.AlgorithmConfig.get_default_rl_module_spec#

AlgorithmConfig.get_default_rl_module_spec() SingleAgentRLModuleSpec | MultiAgentRLModuleSpec[source]#

Returns the RLModule spec to use for this algorithm.

Override this method in the sub-class to return the RLModule spec given the input framework.

Returns:

The RLModuleSpec (SingleAgentRLModuleSpec or MultiAgentRLModuleSpec) to use for this algorithm’s RLModule.