Skip to content

Commit b3603ef

Browse files
committed
fix get lr bug
1 parent 993c057 commit b3603ef

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

src/forge/actors/trainer.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -166,11 +166,7 @@ async def train_step(
166166

167167
t.step("forward_backward")
168168

169-
current_lr = (
170-
self.engine.lr_schedulers.get_last_lr()[0]
171-
if hasattr(self.engine.lr_schedulers, "get_last_lr")
172-
else 0.001
173-
)
169+
current_lr = self.engine.lr_schedulers.schedulers[0].get_last_lr()[0]
174170
record_metric("rl_trainer/learning_rate", current_lr, Reduce.MIN)
175171

176172
self.engine.optimizers.step()

0 commit comments

Comments
 (0)