export
功能描述
Rec SDK训练模型导出接口,导出稀疏表,包括键名和嵌入层表示。以下情况可支持使用export接口导出稀疏表数据。
- 片上内存模式(非动态扩容)
- DDR模式
函数原型
def export(table_list=None):
参数说明
参数名 |
类型 |
可选/必选 |
说明 |
---|---|---|---|
table_list |
list |
可选 |
需要导出的稀疏表列表,不传参默认导出所有稀疏表。 |
返回值说明
- 成功:None。
- 失败:抛出异常。
使用示例
from mx_rec.saver.sparse import export table_list=["sparse_embedding_table"] export(table_list=table_list)
参考资源
接口调用流程及示例,请参见模型迁移与训练。
父主题: 模型接口