Language

unilab.algos.torch.common.actor_factory.build_actor¶

unilab.algos.torch.common.actor_factory.build_actor(algo_type, obs_dim, action_dim, actor_hidden_dim, use_layer_norm, device, num_envs=1, actor_num_blocks=2, actor_noise_zeta_mu=2.0, actor_noise_zeta_max=16, priv_info_dim=None, priv_info_embed_dim=9, priv_mlp_hidden_dims=(256, 128, 9), **kwargs)[source]¶

Build the correct actor model based on algorithm type.

Parameters:

actor_num_blocks (int)
actor_noise_zeta_mu (float)
actor_noise_zeta_max (int)
priv_info_dim (int | None)
priv_info_embed_dim (int)
priv_mlp_hidden_dims (tuple[int, ...] | list[int])