Skip to content
Snippets Groups Projects
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
Branches
Tags
1 merge request!57Updates to the logits estimator
...@@ -78,6 +78,8 @@ def append_image_augmentation(image, ...@@ -78,6 +78,8 @@ def append_image_augmentation(image,
# Casting to float32 # Casting to float32
image = tf.cast(image, tf.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: if output_shape is not None:
assert len(output_shape) == 2 assert len(output_shape) == 2
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment