We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 13a0e34 commit a9b8d8cCopy full SHA for a9b8d8c
1 file changed
src/instructlab/training/main_ds.py
@@ -253,9 +253,7 @@ def _save_and_exit(checkpoint_location: str) -> None:
253
# the check runs after every minibatch backward rather than
254
# waiting for the full optimizer step.
255
_interrupt_check = (
256
- (lambda: check_checkpoint_requested())
257
- if on_demand_checkpointing
258
- else None
+ check_checkpoint_requested if on_demand_checkpointing else None
259
)
260
batch_metrics, avg_loss_across_ranks = batch_loss_manager.process_batch(
261
batch, interrupt_check=_interrupt_check
0 commit comments