Skip to content
Snippets Groups Projects

Resolve "Adopt to the Estimators API"

Merged Tiago de Freitas Pereira requested to merge 40-adopt-to-the-estimators-api into master
2 files
+ 5
4
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -17,16 +17,14 @@ from __future__ import absolute_import
@@ -17,16 +17,14 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import division
from __future__ import print_function
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
import tensorflow as tf
model_dir = '/tmp/mnist_model'
model_dir = '/tmp/mnist_model'
train_tfrecords = ['/tmp/mnist_data/train.tfrecords']
train_tfrecords = ['/tmp/mnist_data/train.tfrecords']
eval_tfrecords = ['/tmp/mnist_data/test.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(keep_checkpoint_max=10**3)
run_config = run_config.replace(save_checkpoints_secs=60)
run_config = run_config.replace(save_checkpoints_secs=60)
Loading