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)
参考资源
接口调用流程及示例,请参见模型迁移与训练。
父主题: 模型接口