The Trainer class automatically handles multi-GPU training, you don’t have to do anything special.
5 Likes
The Trainer class automatically handles multi-GPU training, you don’t have to do anything special.