.

openpsi-project · Sep 3, 2024 · 0c96e81 · 0c96e81
1 parent 0616fee
commit 0c96e81
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/realhf/impl/model/interface/ppo_interface.py b/realhf/impl/model/interface/ppo_interface.py
@@ -396,7 +396,7 @@ def train_step(
             min_size=(
                 constants.pipe_parallel_world_size() * 2 * n_mbs
                 if constants.pipe_parallel_world_size() > 1
-                else constants.pipe_parallel_world_size() * n_mbs
+                else n_mbs
             ),
         )
 
@@ -812,7 +812,7 @@ def train_step(
             min_size=(
                 constants.pipe_parallel_world_size() * 2 * n_mbs
                 if constants.pipe_parallel_world_size() > 1
-                else constants.pipe_parallel_world_size() * n_mbs
+                else n_mbs
             ),
         )