forward_inference#

MultiRLModule.forward_inference(batch: Dict[str, Any], **kwargs) → Dict[str, Any]#

DO NOT OVERRIDE! Forward-pass during evaluation, called from the sampler.

This method should not be overridden. Override the self._forward_inference() method instead.

Parameters:

  • batch (Dict[str, Any]) – The input batch.

  • **kwargs – Additional keyword arguments.

Returns:

The output of the forward pass. This output should comply with the ouptut_specs_inference().

Return type:

Dict[str, Any]