train under multi gpu

[abinlee]

Have you tried to enable multi gpu to train?
If I use “-n_gpu 2” to train model, I get this error:
File "Keras-ICNet/model.py", line 195, in
y = Lambda(lambda x: tf.image.resize_bilinear(x, size=(int(x.shape[1])//2, int(x.shape[2])//2)), name='data_sub2')(x)
TypeError: int returned non-int (type NoneType)

keras 2.1.3
tensorflow 1.4.1