[object Object]

[object Object][object Object]undefined[object Object]

• API function: Implements the AdamW optimizer.

• Formula:

  $$g_t=\begin{cases}-g_t & \text{ if } maximize= true\\ g_t & \text{ if } maximize=false \end{cases}$$ $$m_{t}=\beta_{1} m_{t-1}+\left(1-\beta_{1}\right) g_{t} \\$$ $$v_{t}=\beta_{2} v_{t-1}+\left(1-\beta_{2}\right) g_{t}^{2}$$ $$\beta_{1}^{t}=\beta_{1}^{t-1}\times\beta_{1}$$ $$\beta_{2}^{t}=\beta_{2}^{t-1}\times\beta_{2}$$ $$v_t=\begin{cases}\max(maxGradNorm, v_t) & \text{ if } amsgrad = true\\ v_t & \text{ if } amsgrad = false \end{cases}$$ $$\hat{m}_{t}=\frac{m_{t}}{1-\beta_{1}^{t}} \\$$ $$\hat{v}_{t}=\frac{v_{t}}{1-\beta_{2}^{t}} \\$$ $$\theta_{t+1}=\theta_{t}-\frac{\eta}{\sqrt{\hat{v}_{t}}+\epsilon} \hat{m}_{t}-\eta \cdot \lambda \cdot \theta_{t-1}$$

[object Object]

Each operator has calls. First, aclnnApplyAdamWGetWorkspaceSize is called to obtain the workspace size required for computation and the executor that contains the operator computation process. Then, aclnnApplyAdamW is called to perform computation.

[object Object]

• Parameters:

  [object Object]

• Returns:

  [object Object]: status code. For details, see .

  The first-phase API implements input parameter verification. The following errors may be thrown.

  [object Object]

[object Object]

• Parameters:

  [object Object]

• Returns:

  [object Object]: status code. For details, see .

[object Object]

• Deterministic compute:
  • aclnnApplyAdamW defaults to a deterministic implementation.

[object Object]

The following example is for reference only. For details, see .