openrl.modules.common package¶

Submodules¶

openrl.modules.common.base_net module¶

class openrl.modules.common.base_net.BaseNet[source]¶: Bases: abc.ABC

openrl.modules.common.dqn_net module¶

class openrl.modules.common.dqn_net.DQNNet(env: Union[gymnasium.core.Env, str], cfg=None, device: Union[torch.device, str] = 'cpu', n_rollout_threads: int = 1, model_dict: Optional[Dict[str, Any]] = None)[source]¶

Bases: openrl.modules.common.base_net.BaseNet

act(observation: Union[numpy.ndarray, Dict[str, numpy.ndarray]]) → Tuple[numpy.ndarray, Optional[Tuple[numpy.ndarray, ...]]][source]¶

load_policy(path: str) → None[source]¶

reset(env: Optional[gymnasium.core.Env] = None) → None[source]¶

openrl.modules.common.mat_net module¶

class openrl.modules.common.mat_net.MATNet(env: Union[gymnasium.core.Env, str], cfg=None, device: Union[torch.device, str] = 'cpu', n_rollout_threads: int = 1, model_dict: Dict[str, Any] = {'model': <class 'openrl.modules.networks.MAT_network.MultiAgentTransformer'>})[source]¶: Bases: openrl.modules.common.ppo_net.PPONet

openrl.modules.common.ppo_net module¶

class openrl.modules.common.ppo_net.PPONet(env: Union[gymnasium.core.Env, str], cfg=None, device: Union[torch.device, str] = 'cpu', n_rollout_threads: int = 1, model_dict: Optional[Dict[str, Any]] = None, module_class: openrl.modules.base_module.BaseModule = <class 'openrl.modules.ppo_module.PPOModule'>)[source]¶

Bases: openrl.modules.common.base_net.BaseNet

act(observation: Union[numpy.ndarray, Dict[str, numpy.ndarray]], available_actions: Optional[numpy.ndarray] = None, deterministic: bool = False) → Tuple[numpy.ndarray, Optional[Tuple[numpy.ndarray, ...]]][source]¶

load_policy(path: str) → None[source]¶

reset(env: Optional[gymnasium.core.Env] = None) → None[source]¶

openrl.modules.common.vdn_net module¶

class openrl.modules.common.vdn_net.VDNNet(env: Union[gymnasium.core.Env, str], cfg=None, device: Union[torch.device, str] = 'cpu', n_rollout_threads: int = 1, model_dict: Optional[Dict[str, Any]] = None)[source]¶

Bases: openrl.modules.common.base_net.BaseNet

act(observation: Union[numpy.ndarray, Dict[str, numpy.ndarray]]) → Tuple[numpy.ndarray, Optional[Tuple[numpy.ndarray, ...]]][source]¶

load_policy(path: str) → None[source]¶

reset(env: Optional[gymnasium.core.Env] = None) → None[source]¶

Module contents¶

class openrl.modules.common.BaseNet[source]¶: Bases: abc.ABC

class openrl.modules.common.DQNNet(env: Union[gymnasium.core.Env, str], cfg=None, device: Union[torch.device, str] = 'cpu', n_rollout_threads: int = 1, model_dict: Optional[Dict[str, Any]] = None)[source]¶

Bases: openrl.modules.common.base_net.BaseNet

act(observation: Union[numpy.ndarray, Dict[str, numpy.ndarray]]) → Tuple[numpy.ndarray, Optional[Tuple[numpy.ndarray, ...]]][source]¶

load_policy(path: str) → None[source]¶

reset(env: Optional[gymnasium.core.Env] = None) → None[source]¶

class openrl.modules.common.MATNet(env: Union[gymnasium.core.Env, str], cfg=None, device: Union[torch.device, str] = 'cpu', n_rollout_threads: int = 1, model_dict: Dict[str, Any] = {'model': <class 'openrl.modules.networks.MAT_network.MultiAgentTransformer'>})[source]¶: Bases: openrl.modules.common.ppo_net.PPONet

class openrl.modules.common.PPONet(env: Union[gymnasium.core.Env, str], cfg=None, device: Union[torch.device, str] = 'cpu', n_rollout_threads: int = 1, model_dict: Optional[Dict[str, Any]] = None, module_class: openrl.modules.base_module.BaseModule = <class 'openrl.modules.ppo_module.PPOModule'>)[source]¶

Bases: openrl.modules.common.base_net.BaseNet

act(observation: Union[numpy.ndarray, Dict[str, numpy.ndarray]], available_actions: Optional[numpy.ndarray] = None, deterministic: bool = False) → Tuple[numpy.ndarray, Optional[Tuple[numpy.ndarray, ...]]][source]¶

load_policy(path: str) → None[source]¶

reset(env: Optional[gymnasium.core.Env] = None) → None[source]¶

class openrl.modules.common.VDNNet(env: Union[gymnasium.core.Env, str], cfg=None, device: Union[torch.device, str] = 'cpu', n_rollout_threads: int = 1, model_dict: Optional[Dict[str, Any]] = None)[source]¶

Bases: openrl.modules.common.base_net.BaseNet

act(observation: Union[numpy.ndarray, Dict[str, numpy.ndarray]]) → Tuple[numpy.ndarray, Optional[Tuple[numpy.ndarray, ...]]][source]¶

load_policy(path: str) → None[source]¶

reset(env: Optional[gymnasium.core.Env] = None) → None[source]¶