transformers

call on_train_end when optuna trial is pruned

#16536

Merged