昇腾社区首页
中文
注册

get_dense_and_sparse_variable

功能描述

获取模型中密集层参数变量和稀疏层参数变量。

函数原型

def get_dense_and_sparse_variable()

返回值说明

  • 成功:返回密集层参数变量和稀疏层参数变量。
  • 失败:抛出异常。

使用示例

from mx_rec.util.variable import get_dense_and_sparse_variable
dense_variables, sparse_variables = get_dense_and_sparse_variable()

参考资源

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