ray.rllib.core.rl_module.rl_module.RLModule.input_specs_train#

RLModule.input_specs_train() Union[List[Union[str, Tuple[str, ...]]], NestedDict[Optional[Constraint]]][source]#

Returns the input specs of the forward_train method.