p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix callback

This commit is contained in:
hiyouga 2023-07-15 22:01:43 +08:00
parent 8528a84e74
commit 552d773dad
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/llmtuner/extras/callbacks.py
View File

@ -17,6 +17,7 @@ class LogCallback(TrainerCallback):
def __init__(self, runner=None): def __init__(self, runner=None):
self.runner = runner self.runner = runner
self.start_time = time.time()
self.tracker = {} self.tracker = {}
def on_train_begin(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, **kwargs): def on_train_begin(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, **kwargs):