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

fix callback

Browse Source
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
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):