Model APIs#

ModelV2 API (rllib.env.models.modelv2.ModelV2)#

All RLlib neural network models have to be provided as ModelV2 sub-classes.

RLlib comes with two sub-classes for TF (keras) models and PyTorch models:

TFModelV2 (rllib.env.models.tf.tf_modelv2.TFModelV2)#

TorchModelV2 (rllib.env.models.torch.torch_modelv2.TorchModelV2)#