多GPU时候应该怎样设置

[gzp001015]

您好，我现在正在试验您的代码，是在拥有100W人的数据集上微调一个已经训练好的人脸识别模型，我目前遇到的困惑是：只能使用单张显卡，不知道在多GPU下怎样设置您的TripletLoss层，我尝试用两张显卡，但似乎loss只在第一张上起作用。