StableBaselines-3 DDPG + HER 多重处理
我正在阅读有关 HER 的文档以及 stable-baselines3 网站中有关多重处理的文档 但是,当我尝试训练时,它会抛出错误! 有没有使用 replay_buffer 进行多处理的示例?
num_cpu = 4
env = make_vec_env(env_name, n_envs=num_cpu, seed=0, vec_env_cls=SubprocVecEnv)
model = DDPG(
'MultiInputPolicy',
env,
verbose = 1,
tensorboard_log = log_path,
replay_buffer_class = HerReplayBuffer,
# Parameters for HER
replay_buffer_kwargs=dict(
n_sampled_goal = 4,
goal_selection_strategy = GoalSelectionStrategy.FUTURE,
online_sampling = True,
),
)
model.learn(total_timesteps = 1000000)
ValueError:无法将输入数组从形状 (6,4) 广播到形状 (4,)
I was reading documentation about HER and also about Multiprocessing in stable-baselines3 website
However when i try to train it throws a error!
Is there any example anywhere with multiprocessing with replay_buffer??
num_cpu = 4
env = make_vec_env(env_name, n_envs=num_cpu, seed=0, vec_env_cls=SubprocVecEnv)
model = DDPG(
'MultiInputPolicy',
env,
verbose = 1,
tensorboard_log = log_path,
replay_buffer_class = HerReplayBuffer,
# Parameters for HER
replay_buffer_kwargs=dict(
n_sampled_goal = 4,
goal_selection_strategy = GoalSelectionStrategy.FUTURE,
online_sampling = True,
),
)
model.learn(total_timesteps = 1000000)
ValueError: could not broadcast input array from shape (6,4) into shape (4,)
如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论
评论(1)
稳定基线3 HER 不支持多处理。
Stable baselines3 HER does not support multiprocessing.