稳定基线多输入策略

问题描述

我尝试通过以下方式使用 MultiInputPolicy:

model = PPO("MultiInputPolicy",env,verbose = 1)

但是,我收到一个错误

KeyError: "Error: unkNown policy type MultiInputPolicy,the only registed policy type are: ['MlpPolicy','CnnPolicy']!"

请帮忙。我该如何解决这个问题?

解决方法

您似乎安装了旧版本的 stable-baselines3。如果您已经安装使用

pip install stable-baselines3

,它可能不是 master 分支的最新版本。从最新的主分支安装:

pip install git+https://github.com/DLR-RM/stable-baselines3.git

对于 pipenv,您可能必须指定 egg

pipenv install git+https://github.com/DLR-RM/stable-baselines3.git#egg=stable_baselines3

MultiInputPolicyadded on 5 月 11 日。最新的 pypi 版本 is of 3 月 17 日。