Skip to content

lit_sb3.algorithms.ppo_mask.policies

[docs] module lit_sb3.algorithms.ppo_mask.policies 

1
2
3
4
5
6
7
from lit_sb3.policies.maskable.actor_critic import MaskableActorCriticPolicy
from lit_sb3.policies.maskable.actor_critic_cnn import MaskableActorCriticCnnPolicy
from lit_sb3.policies.maskable.multi_input_actor_critic import MaskableMultiInputActorCriticPolicy

MlpPolicy = MaskableActorCriticPolicy
CnnPolicy = MaskableActorCriticCnnPolicy
MultiInputPolicy = MaskableMultiInputActorCriticPolicy