ray.rllib.policy.torch_policy_v2.TorchPolicyV2.make_model

TorchPolicyV2.make_model() → ray.rllib.models.modelv2.ModelV2

Create model.

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

Returns

ModelV2 model.