Merge pull request cleverhans-lab#249 from fartashf/master

Added the bias to Conv2D in tutorial_mnist_tf
bjtuxck · Aug 25, 2017 · aa05444 · aa05444
2 parents 2074864 + 0727cb2
commit aa05444
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cleverhans_tutorials/tutorial_models.py b/cleverhans_tutorials/tutorial_models.py
@@ -103,8 +103,8 @@ def set_input_shape(self, input_shape):
         self.output_shape = tuple(output_shape)
 
     def fprop(self, x):
-        return tf.nn.conv2d(x, self.kernels,
-                            (1,) + tuple(self.strides) + (1,), self.padding)
+        return tf.nn.conv2d(x, self.kernels, (1,) + tuple(self.strides) + (1,),
+                            self.padding) + self.b
 
 
 class ReLU(Layer):