diff --git a/bob/learn/tensorflow/script/db_to_tfrecords.py b/bob/learn/tensorflow/script/db_to_tfrecords.py index aa5169935749e1022c84cca1a2e3427548a6689d..bfcf50ad2e4ed275c0366a66ada93db243fe8a60 100644 --- a/bob/learn/tensorflow/script/db_to_tfrecords.py +++ b/bob/learn/tensorflow/script/db_to_tfrecords.py @@ -150,14 +150,14 @@ def main(argv=None): logger.info('Processing file %d out of %d', i + 1, n_files) path = f.make_path(data_dir, data_extension) - data = reader(path) + data = reader(path) if data is None: if allow_missing_files: logger.debug("... Processing original data file '{0}' was not successful".format(path)) continue else: raise RuntimeError("Preprocessing of file '{0}' was not successful".format(path)) - + label = file_to_label(f) if one_file_one_sample: diff --git a/bob/learn/tensorflow/script/eval_generic.py b/bob/learn/tensorflow/script/eval_generic.py index ca8bf33a86355af135e063a95d085db2b39c95b4..e8432aa3cb946e29a460dbace75b25908032784e 100644 --- a/bob/learn/tensorflow/script/eval_generic.py +++ b/bob/learn/tensorflow/script/eval_generic.py @@ -63,7 +63,7 @@ def main(argv=None): model_fn = config.model_fn eval_input_fn = config.eval_input_fn - eval_interval_secs = getattr(config, 'eval_interval_secs', 300) + eval_interval_secs = getattr(config, 'eval_interval_secs', 60) run_once = getattr(config, 'run_once', False) run_config = getattr(config, 'run_config', None) model_params = getattr(config, 'model_params', None) @@ -75,7 +75,7 @@ def main(argv=None): nn = tf.estimator.Estimator(model_fn=model_fn, model_dir=model_dir, params=model_params, config=run_config) if name: - real_name = name + '_eval' + real_name = 'eval_' + name else: real_name = 'eval' evaluated_file = os.path.join(nn.model_dir, real_name, 'evaluated') diff --git a/bob/learn/tensorflow/script/train_generic.py b/bob/learn/tensorflow/script/train_generic.py index 11f7d18a421b8c5ef48196ca254116722f8c5138..e60e4917baf349422a83360220faf2722aaed66f 100644 --- a/bob/learn/tensorflow/script/train_generic.py +++ b/bob/learn/tensorflow/script/train_generic.py @@ -66,9 +66,7 @@ def main(argv=None): if run_config is None: # by default create reproducible nets: - from bob.learn.tensorflow.utils.reproducible import session_conf - run_config = tf.estimator.RunConfig() - run_config.replace(session_config=session_conf) + from bob.learn.tensorflow.utils.reproducible import run_config # Instantiate Estimator nn = tf.estimator.Estimator(model_fn=model_fn, model_dir=model_dir, diff --git a/bob/learn/tensorflow/utils/reproducible.py b/bob/learn/tensorflow/utils/reproducible.py index a2f7fb73bd0c67959a59bd6bba6faa1d83067335..b6258c8db6765c27a6e332db1b8cbc8b66e5cc50 100644 --- a/bob/learn/tensorflow/utils/reproducible.py +++ b/bob/learn/tensorflow/utils/reproducible.py @@ -32,9 +32,11 @@ session_conf = tf.ConfigProto(intra_op_parallelism_threads=1, # in the TensorFlow backend have a well-defined initial state. # For further details, see: # https://www.tensorflow.org/api_docs/python/tf/set_random_seed -tf.set_random_seed(1234) +tf_random_seed = 1234 +tf.set_random_seed(tf_random_seed) # sess = tf.Session(graph=tf.get_default_graph(), config=session_conf) # keras.backend.set_session(sess) run_config = tf.estimator.RunConfig() run_config = run_config.replace(session_config=session_conf) +run_config = run_config.replace(tf_random_seed=tf_random_seed)