Closed RaviMa-Kore closed 3 years ago
@mihail-amazon Does this model support multi-gpu training? I see that the code uses only one gpu for training!
The downstream models typically do not need more than a single GPU to train but you're welcome to adapt them to fit your needs.
@mihail-amazon Does this model support multi-gpu training? I see that the code uses only one gpu for training!