ray.rllib.policy.torch_policy_v2.TorchPolicyV2.make_model_and_action_dist
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.