ray.rllib.policy.torch_policy_v2.TorchPolicyV2.optimizer#
- TorchPolicyV2.optimizer() List[torch.optim.Optimizer] | torch.optim.Optimizer[source]#
Custom the local PyTorch optimizer(s) to use.
- Returns:
The local PyTorch optimizer(s) to use for this Policy.