昇腾社区首页
中文
注册

export

功能描述

读取当前step已保存的稀疏表checkpoint,以key-value形式保存为numpy文件,其中key为查询ID,value为嵌入层表示。以下情况可支持使用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)

参考资源

接口调用流程及示例,请参见模型迁移与训练