Hugging Face Forums
Custom model with two pretrained models fails multi GPU training when using the Trainer
🤗Transformers
shamanez
March 2, 2023, 8:59am
1
How can I wrap the custom model class to use the trainer in multiple GPU settings