experiments
/
STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #1780 from mozilla/set-random-seed 

Set graph's random seed when training
This commit is contained in:
Reuben Morais 2018-12-12 10:04:51 -02:00 committed by GitHub
parent da135ca3f9 9ae2e5b3b2
commit f0ba315aeb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
DeepSpeech.py
View File

@ -890,6 +890,7 @@ def main(_):
if len(FLAGS.worker_hosts) == 0: if len(FLAGS.worker_hosts) == 0:
# Only one local task: this process (default case - no cluster) # Only one local task: this process (default case - no cluster)
with tf.Graph().as_default(): with tf.Graph().as_default():
tf.set_random_seed(FLAGS.random_seed)
train() train()
# Now do a final test epoch # Now do a final test epoch
if FLAGS.test: if FLAGS.test:

2
util/flags.py
View File

@ -111,7 +111,7 @@ def create_flags():
# Initialization # Initialization
tf.app.flags.DEFINE_integer ('random_seed', 4567, 'default random seed that is used to initialize variables') tf.app.flags.DEFINE_integer ('random_seed', 4568, 'default random seed that is used to initialize variables')
# Early Stopping # Early Stopping