From 552d773dad0a8e6bcfdd4f4f587e1fe17edcd7b3 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

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