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

[shamanez]

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