Skip to content

Commit 54179be

Browse files
committed
remove confusing log
1 parent fb525eb commit 54179be

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

trinity/trainer/verl/fsdp_workers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -464,7 +464,7 @@ def _build_model_optimizer( # noqa: C901
464464
num_warmup_steps = int(num_warmup_steps_ratio * total_steps)
465465

466466
if self.rank == 0:
467-
print(f"Total steps: {total_steps}, num_warmup_steps: {num_warmup_steps}")
467+
print(f"num_warmup_steps: {num_warmup_steps}")
468468

469469
if warmup_style == "constant":
470470
actor_lr_scheduler = get_constant_schedule_with_warmup(
@@ -1180,7 +1180,7 @@ def _build_critic_model_optimizer(self, config): # noqa: C901
11801180
num_warmup_steps = int(num_warmup_steps_ratio * total_steps)
11811181

11821182
if self.rank == 0:
1183-
print(f"Total steps: {total_steps}, num_warmup_steps: {num_warmup_steps}")
1183+
print(f"num_warmup_steps: {num_warmup_steps}")
11841184

11851185
from verl.utils.torch_functional import (
11861186
get_constant_schedule_with_warmup,

0 commit comments

Comments
 (0)