Class Adam

Inheritance Relationships

Base Type

  • public mindspore::ops::BaseOperator

Class Documentation

class Adam : public mindspore::ops::BaseOperator

Updates gradients by the Adaptive Moment Estimation (Adam) algorithm. Refer to Python API mindspore.ops.Adam for more details.

Public Functions

inline Adam()

Constructor.

void Init(const bool use_locking = false, const bool use_nesterov = false)

Init. Refer to the parameters of Python API mindspore.ops.Adam for the inputs.

void set_use_locking(const bool use_locking)

Set use_locking.

void set_use_nesterov(const bool use_nesterov)

Set use_nesterov.

bool get_use_locking() const

Get use_locking.

返回

use_locking.

bool get_use_nesterov() const

Get use_nesterov.

返回

use_nesterov.