Commit 896d0772 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Forcing the seed of the input_fn random operations

parent 23573ef0
......@@ -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
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment