ray.rllib.core.rl_module.marl_module.MultiAgentRLModule.forward_train

MultiAgentRLModule.forward_train(batch: Union[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 ouptut_specs_train().