下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

动态Shape

mxRec训练框架支持动态Shape,用户可参考使用流程启用动态Shape功能。

启用动态Shape功能前,需要安装Kernels算子包,具体操作请参见《CANN 软件安装指南》的“安装CANN(物理机)”章节的“安装Kernels”部分。

使用流程介绍

如需在mxRec训练框架使用动态Shape功能,可在初始化框架时,配置init接口的“use_dynamic”参数为“True”

示例代码

init(use_mpi, rank_id=rank_id, rank_size=rank_size, train_steps=train_steps, eval_steps=eval_steps, prefetch_batch_number=1, use_dynamic=True)
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词