trainer

tfts Trainer

Classes

BaseTrainer(model[, args, strategy])	Trainer for pipeline
KerasTrainer(model[, strategy, args])	Keras trainer from tf.keras
Seq2seqKerasTrainer(*args, **kwargs)	As the transformers forum mentioned: https://discuss.huggingface.co/t/trainer-vs-seq2seqtrainer/3145/2 Seq2SeqTrainer is mostly about predict_with_generate.
Trainer(model[, strategy])	Custom trainer for tensorflow with support for CPU, GPU, and multi-GPU.