ray.rllib.core.rl_module.rl_module.RLModule.output_specs_train#

RLModule.output_specs_train() List[str | Tuple[str, ...]] | NestedDict[Constraint | None][source]#

Returns the output specs of the forward_train method.