riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 10:12:23 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Add kwargs to CallbackHandler.on_evaluate signature

Browse Source
This commit is contained in:
Lewis Tunstall 2021-04-20 23:16:29 +02:00
parent 9bb6e0d602
commit 8d4c85ed51
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/trainer_callback.py
View File

@ -358,9 +358,9 @@ class CallbackHandler(TrainerCallback):
def on_step_end(self, args: TrainingArguments, state: TrainerState, control: TrainerControl):
return self.call_event("on_step_end", args, state, control)
def on_evaluate(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, metrics):
def on_evaluate(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, metrics, **kwargs):
control.should_evaluate = False
return self.call_event("on_evaluate", args, state, control, metrics=metrics)
return self.call_event("on_evaluate", args, state, control, metrics=metrics, **kwargs)
def on_save(self, args: TrainingArguments, state: TrainerState, control: TrainerControl):
control.should_save = False