Как исправить ошибку Concatenate при использовании multi_gpu_model в керасе?

Question

Я пытаюсь тренировать модель keras с несколькими gpus, но при компиляции я получил ошибку Concatenate ， при использовании одного gpu работает нормально.что я должен делать?

Я использую tensorflow=1.4.0, keras=2.1.3 и python=3.5.6

Я установил gpu на 1, используя os.environ["CUDA_VISIBLE_DEVICES"] = "0", и он работает, но при использовании os.environ["CUDA_VISIBLE_DEVICES"] = "0,1" он не сможетcompile.

Код, который я использовал для тестирования:

 from utils.model.SEResNeXt import SEResNeXt
 from keras.utils.training_utils import multi_gpu_model
 import os
 #os.environ['CUDA_VISIBLE_DEVICES'] = '0'
 os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'
 model = SEResNeXt(size=96).model
 model = multi_gpu_model(model,2)

 model.summary()

, и код, который я использовал для построения модели, находится в этой ссылке (спасибо yoheikikuta агрегат)

ValueError: A `Concatenate` layer requires inputs with matching shapes except for the concat axis. Got inputs shapes: [(None, None, None, 32), (None, 48, 48, 64), (None, None, None, 32)]