昇腾社区首页
中文
注册

(beta)torch_npu.contrib.module.NpuCachedDropout

函数原型

[object Object]

功能说明

在NPU设备上使用FairseqDropout。

参数说明

  • p (Float) - 元素归零的概率。
  • module_name (String) - 模型名称。

支持的型号

  • [object Object]Atlas 训练系列产品[object Object]
  • [object Object]Atlas A2 训练系列产品[object Object]
  • [object Object]Atlas A3 训练系列产品[object Object]
  • [object Object]Atlas 推理系列产品[object Object]