Skip to content

GitLab

Commit 7b6c8999 authored by guptapriya's avatar guptapriya Committed by guptapriya
Browse files

set tf seed

parent c80de2ca
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
official/recommendation/ncf_keras_main.py
View file @ 7b6c8999
...@@ -245,6 +245,10 @@ def _get_keras_model(params): ...@@ -245,6 +245,10 @@ def _get_keras_model(params):
def run_ncf(_): def run_ncf(_):
if FLAGS.seed is not None:
print("Setting tf seed")
tf.random.set_seed(FLAGS.seed)
"""Run NCF training and eval with Keras.""" """Run NCF training and eval with Keras."""
# TODO(seemuch): Support different train and eval batch sizes # TODO(seemuch): Support different train and eval batch sizes
if FLAGS.eval_batch_size != FLAGS.batch_size: if FLAGS.eval_batch_size != FLAGS.batch_size:
......
Markdown is supported
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