reference/generated/torchrl.trainers.algorithms.configs.modules.ModelConfig

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

torchrl.trainers.algorithms.configs.modules.ModelConfig

class torchrl.trainers.algorithms.configs.modules.ModelConfig(_partial_: bool = False, in_keys: Any = None, out_keys: Any = None, shared: bool = False)

Parent class to configure a model.

A model can be made of several networks. It is always a TensorDictModuleBase instance.

See also

TanhNormalModelConfig, ValueModelConfig