ray.rllib.core.rl_module.rl_module.RLModule.forward_exploration#

RLModule.forward_exploration(batch: Union[SampleBatch, MultiAgentBatch], **kwargs) → Mapping[str, Any][source]#

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 ouptut_specs_exploration().