fix callback

Former-commit-id: 477ef5ffd84c78ab1a8bce27714bb4f6e6ca0210
2025-10-14 23:58:11 +08:00 · 2023-07-15 22:01:43 +08:00 · 2023-07-15 22:01:43 +08:00 · f8831cb1ea
commit f8831cb1ea
parent 6a0499ef40
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):