Hot_Embedding
Hot_Embedding功能默认开启，无需配置。
示例代码
该功能自动开启，无需配置；为了确保使能成功，需要开启GatherV2算子的高性能模式，开启方法如下：
- 将op_impl_mode.ini配置项传入到Session。代码如下：
1 2 3 4 5 6 7 8 9 10
import tensorflow as tf session_config = tf.compat.v1.ConfigProto(allow_soft_placement=False, log_device_placement=False) session_config.gpu_options.allow_growth = True custom_op = session_config.graph_options.rewrite_options.custom_optimizers.add() # 1. 将算子配置文件路径传入配置项 custom_op.parameter_map["op_precision_mode"].s = tf.compat.as_bytes("op_impl_mode.ini") # 2. 建图 # 3. 将config配置传入sess初始化中 with tf.compat.v1.Session(config=sess_config) as sess: # 4. 训练
- 在模型的路径下创建op_impl_mode.ini文件，内容如下：
1GatherV2=high_performance
父主题： 训练功能特性流程