ray.rllib.policy.torch_policy_v2.TorchPolicyV2.get_session

TorchPolicyV2.get_session() → tensorflow.compat.v1.Session | None

Returns tf.Session object to use for computing actions or None.

Note: This method only applies to TFPolicy sub-classes. All other sub-classes should expect a None to be returned from this method.

Returns:

The tf Session to use for computing actions and losses with

this policy or None.