mirror of
https://github.com/saymrwulf/stable-baselines3.git
synced 2026-06-06 00:03:28 +00:00
23afedb254
* Add auto formatting with black and isort * Reformat code * Ignore typing errors * Add note about line length * Add minimum version for isort * Add commit-checks * Update docker image * Fixed lost import (during last merge) * Fix opencv dependency
9 lines
342 B
Python
9 lines
342 B
Python
# This file is here just to define MlpPolicy/CnnPolicy
|
|
# that work for PPO
|
|
from stable_baselines3.common.policies import ActorCriticCnnPolicy, ActorCriticPolicy, register_policy
|
|
|
|
MlpPolicy = ActorCriticPolicy
|
|
CnnPolicy = ActorCriticCnnPolicy
|
|
|
|
register_policy("MlpPolicy", ActorCriticPolicy)
|
|
register_policy("CnnPolicy", ActorCriticCnnPolicy)
|