无法使用亚当优化器运行tensorflow multi-gpu

Question

我正在使用以下代码在本地运行Tensorflow：

display:flex

模型函数使用$('#test').attr('target', '_blank'); 作为优化器。 由于以下原因，火车呼叫失败：

    distribution = tf.contrib.distribute.OneDeviceStrategy(device='/cpu:0')
    config = tf.estimator.RunConfig(train_distribute=distribution)
    estimator_model_fn = create_model_fn()
    estimator = tf.estimator.Estimator(estimator_model_fn, config=config)
    estimator.train(self.train_input_fn,steps=1000)

如果我将优化器更改为sgd，则程序运行将没有任何问题。 有关如何解决此问题的任何想法？