ray.rllib.policy.torch_policy_v2.TorchPolicyV2.extra_compute_grad_fetches

TorchPolicyV2.extra_compute_grad_fetches() → Dict[str, Any]

Extra values to fetch and return from compute_gradients().

Returns:

Extra fetch dict to be added to the fetch dict of the compute_gradients call.