ray.rllib.core.rl_module.rl_module.RLModule.output_specs_inference#
- RLModule.output_specs_inference() List[str | Tuple[str, ...]] [source]#
Returns the output specs of the
forward_inference()
method.Override this method to customize the output specs of the inference call. The default implementation requires the
forward_inference()
method to return a dict that hasaction_dist
key and its value is an instance ofDistribution
.