Skip to content

lit_sb3.algorithms.a2c.policies

[docs] module lit_sb3.algorithms.a2c.policies 

1
2
3
4
5
6
7
# This file is here just to define MlpPolicy/CnnPolicy
# that work for A2C
from lit_sb3.policies import ActorCriticCnnPolicy, ActorCriticPolicy, MultiInputActorCriticPolicy

MlpPolicy = ActorCriticPolicy
CnnPolicy = ActorCriticCnnPolicy
MultiInputPolicy = MultiInputActorCriticPolicy