ray.rllib.core.rl_module.multi_rl_module.MultiRLModule.forward_exploration#
- MultiRLModule.forward_exploration(batch: SampleBatch | MultiAgentBatch | Dict[str, Any], **kwargs) Dict[str, Any] #
Forward-pass during exploration, called from the sampler.
This method should not be overriden to implement a custom forward exploration method. Instead, override the _forward_exploration method.
- Parameters:
batch – The input batch. This input batch should comply with input_specs_exploration().
**kwargs – Additional keyword arguments.
- Returns:
The output of the forward pass. This output should comply with the output_specs_exploration().