tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Add callbacks to Trainer

Browse Source
This commit is contained in:
NanoCode012
2023-05-09 00:41:19 +09:00
parent 0d6708bfe4
commit cc77bab526
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/axolotl/utils/trainer.py
View File

@@ -204,6 +204,7 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer):
return_tensors="pt", return_tensors="pt",
**data_collator_kwargs, **data_collator_kwargs,
), ),
callbacks=callbacks,
**trainer_kwargs, **trainer_kwargs,
) )