ray.rllib.core.rl_module.marl_module.MultiAgentRLModule.forward_train#
- MultiAgentRLModule.forward_train(batch: SampleBatch | MultiAgentBatch, **kwargs) Mapping[str, Any] #
Forward-pass during training called from the learner. This method should not be overriden. Instead, override the _forward_train method.
- Parameters:
batch – The input batch. This input batch should comply with input_specs_train().
**kwargs – Additional keyword arguments.
- Returns:
The output of the forward pass. This output should comply with the output_specs_train().