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

AlgorithmConfig.get_default_rl_module_spec() → Union[ray.rllib.core.rl_module.rl_module.SingleAgentRLModuleSpec, ray.rllib.core.rl_module.marl_module.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 ModuleSpec (SingleAgentRLModuleSpec or MultiAgentRLModuleSpec) to use for this algorithm’s RLModule.