StableBaselines-3 DDPG + HER 多重处理

发布于 2025-01-15 22:42:15 字数 635 浏览 3 评论 0原文

我正在阅读有关 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 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(1

稚气少女 2025-01-22 22:42:15

稳定基线3 HER 不支持多处理

Stable baselines3 HER does not support multiprocessing.

~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文