打开 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 库时（我slim.learning.train用来开始训练），我不确定何时/如何启用 XLA JIT 。
slim.learning.train接受一个session_config参数，您可以将ConfigProto.
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句