ApplyMomentum.Options

kelas statis publik ApplyMomentum.Options

Atribut opsional untuk ApplyMomentum

Metode Publik

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

Metode Warisan

Metode Publik

ApplyMomentum publik. Opsi useLocking (penguncian penggunaan Boolean)

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 ApplyMomentum.Options 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.