From 896d077201b49e91f6490434cf763fdb55e265ba Mon Sep 17 00:00:00 2001 From: Tiago Freitas Pereira <tiagofrepereira@gmail.com> Date: Tue, 10 Jul 2018 11:52:46 +0200 Subject: [PATCH] Forcing the seed of the input_fn random operations --- bob/learn/tensorflow/dataset/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/bob/learn/tensorflow/dataset/__init__.py b/bob/learn/tensorflow/dataset/__init__.py index 7c0af9b4..a742b1e0 100644 --- a/bob/learn/tensorflow/dataset/__init__.py +++ b/bob/learn/tensorflow/dataset/__init__.py @@ -78,6 +78,8 @@ def append_image_augmentation(image, # Casting to float32 image = tf.cast(image, tf.float32) + # FORCING A SEED FOR THE RANDOM OPERATIONS + tf.set_random_seed(0) if output_shape is not None: assert len(output_shape) == 2 -- GitLab