机器学习PAI为什么在一个session run中有两次GPU呢?

16b9d5f5f6565776e2f7b57908bfe407.png 机器学习PAI为什么在一个session run中有两次LookupForwardEmbeddingVarGPU呢?

展开
收起
冰激凌甜筒 2023-05-22 19:58:05 262 分享 版权
1 条回答
写回答
取消 提交回答
  • 特征列的dim不同会有不同的lookup,为了确保算子的高性能,需要dim相同才会被聚合到同一个的lookup里,此回答整理自钉群“DeepRec用户群”

    2023-05-22 20:14:12
    赞同 展开评论

人工智能平台 PAI(Platform for AI,原机器学习平台PAI)是面向开发者和企业的机器学习/深度学习工程平台,提供包含数据标注、模型构建、模型训练、模型部署、推理优化在内的AI开发全链路服务,内置140+种优化算法,具备丰富的行业场景插件,为用户提供低门槛、高性能的云原生AI工程化能力。

还有其他疑问?
咨询AI助理