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