Passing arguments to tf.config?

tensorflow

(Tony Reina) #1

I’d like to pass the TensorFlow intraop-threads and interop-threads arguments to the session configuration. It’d be great if there was a command line switch to do this. For example,

# Optimize CPU threads for TensorFlow
config = tf.ConfigProto(
		inter_op_parallelism_threads=args.interop_threads,
		intra_op_parallelism_threads=args.intraop_threads)

sess = tf.Session(config=config)

(Guillaume Klein) #2

I just pushed a new version that contains this feature:


(Tony Reina) #3

Ah! I missed that one. Thanks so much. This is wonderful.

Best.
-Tony