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

TorchPolicyV2.make_model_and_action_dist() → Tuple[ray.rllib.models.modelv2.ModelV2, Type[ray.rllib.models.torch.torch_action_dist.TorchDistributionWrapper]][source]#

Create model and action distribution function.

Returns

ModelV2 model. ActionDistribution class.