ray.rllib.policy.torch_policy_v2.TorchPolicyV2.make_rl_module#
- TorchPolicyV2.make_rl_module() RLModule #
Returns the RL Module (only for when RLModule API is enabled.)
If RLModule API is enabled (self.config.api_stack(enable_rl_module_and_learner=True), this method should be implemented and should return the RLModule instance to use for this Policy. Otherwise, RLlib will error out.