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

RLModule.input_specs_train() → List[str | Tuple[str, ...]]

Returns the input specs of the forward_train method.