Custom model with two pretrained models fails multi GPU training when using the Trainer

🤗Transformers
1

How can I wrap the custom model class to use the trainer in multiple GPU settings

1 Like