OpenNMT Forum

Passing arguments to tf.config?

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(

sess = tf.Session(config=config)

I just pushed a new version that contains this feature:

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