423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-15 16:18:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix callback

Browse Source 
Former-commit-id: 477ef5ffd84c78ab1a8bce27714bb4f6e6ca0210
This commit is contained in:
hiyouga 2023-07-15 22:01:43 +08:00
parent 6a0499ef40
commit f8831cb1ea
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):