ray.rllib.policy.torch_policy_v2.TorchPolicyV2.on_global_var_update

TorchPolicyV2.on_global_var_update(global_vars: Dict[str, Union[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.