SparseApplyMomentum.Options

kelas statis publik SparseApplyMomentum.Options

Atribut opsional untuk SparseApplyMomentum

Metode Publik

SparseApplyMomentum.Opsi
useLocking (penguncian penggunaan Boolean)
SparseApplyMomentum.Opsi
useNesterov (Boolean useNesterov)

Metode Warisan

Metode Publik

public SparseApplyMomentum.Options useLocking (Boolean useLocking)

Parameter
gunakan Penguncian Jika `True`, pembaruan tensor var dan accum akan dilindungi oleh kunci; jika tidak, perilaku tersebut tidak terdefinisikan, namun mungkin menunjukkan lebih sedikit pertentangan.

publik SparseApplyMomentum.Opsi useNesterov (Boolean useNesterov)

Parameter
gunakanNesterov Jika `True`, tensor yang diteruskan ke komputasi grad adalah var - lr * momentum * accum, jadi pada akhirnya, var yang Anda dapatkan sebenarnya adalah var - lr * momentum * accum.