Merge pull request #47 from openai/tfseed

added TF seed to both tutorials for reproducibility
cleverhans-lab · Nov 22, 2016 · 5c6ece8 · 5c6ece8
2 parents b1169ac + 717e672
commit 5c6ece8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/tutorials/mnist_tutorial.py b/tutorials/mnist_tutorial.py
@@ -27,6 +27,10 @@ def main(argv=None):
     MNIST cleverhans tutorial
     :return:
     """
+
+    # Set TF random seed to improve reproducibility
+    tf.set_random_seed(1234)
+
     # Image dimensions ordering should follow the Theano convention
     if keras.backend.image_dim_ordering() != 'th':
         keras.backend.set_image_dim_ordering('th')

diff --git a/tutorials/mnist_tutorial_jsma.py b/tutorials/mnist_tutorial_jsma.py
@@ -37,6 +37,9 @@ def main(argv=None):
     :return:
     """
 
+    # Set TF random seed to improve reproducibility
+    tf.set_random_seed(1234)
+
     ###########################################################################
     # Define the dataset and model
     ###########################################################################