tensorflow配置Mask-RCNN报错：tf.reduce_mean got an unexpected keyword argument 'keep_dims'

我的配置环境是：

• Tensorflow-gpu1.4.0
• GTX1080ti
• CUDA8.0&Cudnn6.0
• keras2.1

• python3.5

  在这样的环境中配置好了之后进行训练，在执行下面模块的时候出现报错

model.train(dataset_train, dataset_val, 
            learning_rate=config.LEARNING_RATE, 
            epochs=1, 
            layers='heads')

# Fine tune all layers
model.train(dataset_train, dataset_val, 
            learning_rate=config.LEARNING_RATE / 10,
            epochs=2, 
            layers="all")

报错信息

~/anaconda2/envs/tf/lib/python3.5/site-packages/mask_rcnn-