Set graph's random seed when training

2018-12-11 21:49:19 -02:00 · 2018-12-11 21:49:19 -02:00 · 8ebfe80dd4
parent da135ca3f9
commit 8ebfe80dd4
1 changed files with 1 additions and 0 deletions
--- a/DeepSpeech.py
+++ b/DeepSpeech.py
@ -890,6 +890,7 @@ def main(_):
        if len(FLAGS.worker_hosts) == 0:
            # Only one local task: this process (default case - no cluster)
            with tf.Graph().as_default():
+                tf.set_random_seed(FLAGS.random_seed)
                train()
            # Now do a final test epoch
            if FLAGS.test: