十次梯度优化的慢性能

Question

我在使用上有困难tf.吸附剂在张量流中。你知道吗

代码如下：

dis_diff = base_discriminator(interpolates, self.num_planes, reuse=True,
                                  scope=self.scope, only_gan=True)
gradients = tf.gradients(dis_diff, interpolates)[0]
#this code run for an hour
train_op=tf.train.AdamOptimizer(1e-6).minimize(tf.reduce_mean(gradients))

基本单元鉴别器是resnet-50。运行最新的句子大约需要1个小时。代码来自wgan gp。 我用错了吗？你知道吗