fix callback

Former-commit-id: 552d773dad0a8e6bcfdd4f4f587e1fe17edcd7b3
2025-08-22 22:02:51 +08:00 · 2023-07-15 22:01:43 +08:00 · 2023-07-15 22:01:43 +08:00 · 20e4bf3b5b
commit 20e4bf3b5b
parent 22e979c482
1 changed files with 1 additions and 0 deletions
--- a/src/llmtuner/extras/callbacks.py
+++ b/src/llmtuner/extras/callbacks.py
@ -17,6 +17,7 @@ class LogCallback(TrainerCallback):

    def __init__(self, runner=None):
        self.runner = runner
+        self.start_time = time.time()
        self.tracker = {}

    def on_train_begin(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, **kwargs):