ray.rllib.algorithms.algorithm_config.AlgorithmConfig.get_default_rl_module_spec
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.