Note
Ray 2.10.0 introduces the alpha stage of RLlib’s “new API stack”. The team is currently transitioning algorithms, example scripts, and documentation to the new code base throughout the subsequent minor releases leading up to Ray 3.0.
See here for more details on how to activate and use the new API stack.
Catalog API#
Basic usage#
Use the following basic API to get a default encoder
or action distribution
out of Catalog. To change the catalog behavior, modify the following methods.
Algorithm-specific implementations of Catalog have additional methods,
for example, for building heads
.
Describes the sub-module-architectures to be used in RLModules. |
|
Builds the encoder. |
|
Get the action distribution class. |
|
Returns a tokenizer config for the given space. |
Advanced usage#
The following methods and attributes are used internally by the Catalog to build the default models. Only override them when you need more granular control.
Returns the latent dimensions of the encoder. |
|
Decision tree hook for subclasses to override. |
|
Returns an EncoderConfig for the given input_space and model_config_dict. |
|
Returns a distribution class for the given action space. |