Is really Trainer class support TPU for faster training?


I got this one. Please give me some examples if really possible.

we don’t anymore have a Trainer for Tensorflow, but yeah the PyTorch Trainer supports TPU.

Or you can also use huggingface/accelerate.

1 Like

Can you give me an example on pytorch trainer?