utils.callbacks.generation
utils.callbacks.generation
Callback for generating samples during SFT/Pretrain training.
Classes
| Name | Description |
|---|---|
| SFTGenerationCallback | Callback for generating samples during SFT/Pretrain training. |
SFTGenerationCallback
utils.callbacks.generation.SFTGenerationCallback(trainer)Callback for generating samples during SFT/Pretrain training.
Methods
| Name | Description |
|---|---|
| on_evaluate | Generate samples at specified intervals. |
on_evaluate
utils.callbacks.generation.SFTGenerationCallback.on_evaluate(
args,
state,
control,
**kwargs,
)Generate samples at specified intervals.