Deep_discriminator - missed relu (#1720)

tensorflow · Oct 3, 2019 · 8be915a · 8be915a
1 parent 1843c72
commit 8be915a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensor2tensor/layers/common_layers.py b/tensor2tensor/layers/common_layers.py
@@ -3724,7 +3724,7 @@ def double_discriminator(x, filters1=128, filters2=None,
       tf.reshape(net, [batch_size, -1])
     net = tf.nn.relu(net)
     net = layers().Conv2D(
-        filters2, kernel_size, strides=strides, padding="SAME", name="conv2")(x)
+        filters2, kernel_size, strides=strides, padding="SAME", name="conv2")(net)
     if pure_mean:
       net2 = tf.reduce_mean(net, [1, 2])
     else: