reference/generated/torchrl.modules.CrossCriticGroupSpec

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

CrossCriticGroupSpec

class torchrl.modules.CrossCriticGroupSpec(obs_dim: int, n_agents: int, obs_key: NestedKey, value_key: NestedKey)

Specification for one agent group used by CrossGroupCritic.

Parameters:

• obs_dim (int) – dimensionality of each agent’s observation vector.

• n_agents (int) – number of agents in the group.

• obs_key (NestedKey) – tensordict key holding this group’s observations, e.g. ("soldiers", "observation").

• value_key (NestedKey) – tensordict key where this group’s state values will be written, e.g. ("soldiers", "state_value").