ray.rllib.policy.torch_policy_v2.TorchPolicyV2.optimizer

TorchPolicyV2.optimizer() → Union[List[torch.optim.Optimizer], torch.optim.Optimizer]

Custom the local PyTorch optimizer(s) to use.

Returns

The local PyTorch optimizer(s) to use for this Policy.