fix callback

2023-07-15 22:01:43 +08:00 · 2023-07-15 22:01:43 +08:00 · 552d773dad
parent 8528a84e74
commit 552d773dad
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):