Sophie Qiao

Did you use a trainer from transformers.trainer? If so, you can write class GlobalStepUpdaterCallback(TrainerCallback): def on_step_end(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, **kwargs): model = kwargs.get('model', None) if model and...