hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[RLlib] Discussion 6432: Automatic train_batch_size calculation fix. (#25621)

Browse source
This commit is contained in:
Artur Niederfahrenhorst 2022-06-10 12:15:57 +02:00 • committed by GitHub
parent 67b2eca6a2
commit 8af9ef8fee
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
rllib/algorithms/ppo/ppo.py
View file

@ -11,6 +11,7 @@ Detailed documentation: https://docs.ray.io/en/master/rllib-algorithms.html#ppo
import logging
from typing import List, Optional, Type, Union
import math
from ray.util.debug import log_once
from ray.rllib.agents.trainer import Trainer
@ -325,8 +326,9 @@ class PPO(Trainer):
config["train_batch_size"] > 0
and config["train_batch_size"] % calculated_min_rollout_size != 0
):
new_rollout_fragment_length = config["train_batch_size"] // (
num_workers * config["num_envs_per_worker"]
new_rollout_fragment_length = math.ceil(
config["train_batch_size"]
/ (num_workers * config["num_envs_per_worker"])
)
logger.warning(
"`train_batch_size` ({}) cannot be achieved with your other "