从tf.slim启用XLA JIT

Question

打开Tensorflow XLA JIT的一种方法是使用tf.OptimizerOptions.ON_1标志，将其传递给TF会话，类似于python中的以下行：

config = tf.ConfigProto()
config.graph_options.optimizer_options.global_jit_level = tf.OptimizerOptions.ON_1  
sess = tf.Session(config=config)

但是，当我使用Slim库时，我不确定何时/如何启用XLA JIT（我使用slim.learning.train开始训练）。

Answer 1

slim.learning.train会将session_config个参数传递给ConfigProto。