ray.rllib.policy.torch_policy_v2.TorchPolicyV2.make_model#

TorchPolicyV2.make_model() → ModelV2[source]#

Create model.

Note: only one of make_model or make_model_and_action_dist can be overridden.

Returns:: ModelV2 model.