Skip to content
Snippets Groups Projects
Commit 422a72a6 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Create a default runconfig

parent 882d1245
Branches
Tags
3 merge requests!22Add a prediction script,!21Resolve "Adopt to the Estimators API",!20Create a default runconfig
Pipeline #
......@@ -17,16 +17,14 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from bob.learn.tensorflow.utils.reproducible import session_conf
# create reproducible nets:
from bob.learn.tensorflow.utils.reproducible import run_config
import tensorflow as tf
model_dir = '/tmp/mnist_model'
train_tfrecords = ['/tmp/mnist_data/train.tfrecords']
eval_tfrecords = ['/tmp/mnist_data/test.tfrecords']
# by default create reproducible nets:
run_config = tf.estimator.RunConfig()
run_config = run_config.replace(session_config=session_conf)
run_config = run_config.replace(keep_checkpoint_max=10**3)
run_config = run_config.replace(save_checkpoints_secs=60)
......
......@@ -35,3 +35,6 @@ session_conf = tf.ConfigProto(intra_op_parallelism_threads=1,
tf.set_random_seed(1234)
# 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)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment