Changing the number of threads in TensorFlow on Cifar10
To configure this value, you can pass a tf.ConfigProto
argument when constructing the tf.Session
:
NUM_THREADS = …sess = tf.Session(config=tf.ConfigProto( intra_op_parallelism_threads=NUM_THREADS))