ray.rllib.policy.torch_policy_v2.TorchPolicyV2.make_model_and_action_dist#

TorchPolicyV2.make_model_and_action_dist() Tuple[ModelV2, Type[TorchDistributionWrapper]][source]#

Create model and action distribution function.

Returns:

ModelV2 model. ActionDistribution class.