Set graph's random seed when training

This commit is contained in:
Reuben Morais 2018-12-11 21:49:19 -02:00 коммит произвёл GitHub
Родитель da135ca3f9
Коммит 8ebfe80dd4
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 0 удалений

Просмотреть файл

@ -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: