IvanaXu / iDeepRec

DeepRec For Me https://github.com/alibaba/DeepRec
https://deeprec.readthedocs.io/zh/latest/index.html
Apache License 2.0
3 stars 1 forks source link

part1 🌈 w0.6 sess_config.* #43

Closed IvanaXu closed 1 year ago

IvanaXu commented 1 year ago

https://deeprec.readthedocs.io/zh/latest/Smart-Stage.html

  • sess_config.graph_options.optimizer_options.do_smart_stage = True

https://deeprec.readthedocs.io/zh/latest/Auto-Fusion.html

  • sess_config.graph_options.optimizer_options.do_op_fusion = True

https://deeprec.readthedocs.io/zh/latest/Async-Embedding-Stage.html

  • sess_config.graph_options.optimizer_options.do_async_embedding = True
  • sess_config.graph_options.optimizer_options.async_embedding_threads_num = 4
  • sess_config.graph_options.optimizer_options.async_embedding_capacity = 4

https://deeprec.readthedocs.io/zh/latest/Executor-Optimization.html

  • sess_config.executor_policy = tf.ExecutorPolicy.USE_INLINE_EXECUTOR

https://deeprec.readthedocs.io/zh/latest/XLA.html

  • sess_config.graph_options.optimizer_options.global_jit_level = tf.OptimizerOptions.ON_1