We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 993c057 commit b3603efCopy full SHA for b3603ef
src/forge/actors/trainer.py
@@ -166,11 +166,7 @@ async def train_step(
166
167
t.step("forward_backward")
168
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
- )
+ current_lr = self.engine.lr_schedulers.schedulers[0].get_last_lr()[0]
174
record_metric("rl_trainer/learning_rate", current_lr, Reduce.MIN)
175
176
self.engine.optimizers.step()
0 commit comments