Skip to content

lit_sb3.algorithms.trpo.policies

[docs] module lit_sb3.algorithms.trpo.policies 

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

MlpPolicy = ActorCriticPolicy
CnnPolicy = ActorCriticCnnPolicy
MultiInputPolicy = MultiInputActorCriticPolicy