From f8831cb1ea7498965b625e1f557c6b267725d5af Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Sat, 15 Jul 2023 22:01:43 +0800
Subject: [PATCH] fix callback

Former-commit-id: 477ef5ffd84c78ab1a8bce27714bb4f6e6ca0210
---
 src/llmtuner/extras/callbacks.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/llmtuner/extras/callbacks.py b/src/llmtuner/extras/callbacks.py
index 351b7cf6..493596be 100644
--- 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):