ray.rllib.policy.torch_policy_v2.TorchPolicyV2.on_global_var_update

TorchPolicyV2.on_global_var_update(global_vars: Dict[str, numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor]) → None

Called on an update to global vars.

Parameters:

global_vars – Global variables by str key, broadcast from the driver.