fix callback
This commit is contained in:
1 parent
8528a84e74
commit
552d773dad
1 file changed
+1
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user