Catalog API
Contents
Note
This doc is related to the RLModule API and therefore experimental.
Catalog API#
Basic usage#
Use the following basic API to get a default encoder
or action distribution
out of Catalog. You can inherit from Catalog and modify the following methods to
directly inject custom components into a given RLModule.
Algorithm-specific implementations of Catalog have additional methods,
for example, for building heads
.
|
Describes the sub-modules architectures to be used in RLModules. |
|
Builds the encoder. |
|
Get the action distribution class. |
|
Returns a suitable preprocessor for the given observation space. |
Advanced usage#
The following methods are used internally by the Catalog to build the default models.
Returns the latent dimensions of the encoder. |
|
Post-init hook for subclasses to override. |
|
|
Returns an EncoderConfig for the given input_space and model_config_dict. |
|
Returns a tokenizer config for the given space. |