Perbarui '*var' sesuai dengan skema momentum.
Tetapkan use_nesterov = True jika Anda ingin menggunakan momentum Nesterov.
accum = accum * momentum - lr * lulusan var += accum
Kelas Bersarang
kelas | ResourceApplyKerasMomentum.Options | Atribut opsional untuk ResourceApplyKerasMomentum |
Metode Publik
statis <T> ResourceApplyKerasMomentum | |
ResourceApplyKerasMomentum.Options statis | useLocking (penguncian penggunaan Boolean) |
ResourceApplyKerasMomentum.Options statis | useNesterov (Boolean useNesterov) |
Metode Warisan
Metode Publik
public static ResourceApplyKerasMomentum buat ( Ruang lingkup , Operan <?> var, Operan <?> accum, Operan <T> lr, Operan <T> grad, Operan <T> momentum, Opsi... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi ResourceApplyKerasMomentum baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
var | Harus dari Variabel(). |
akumulasi | Harus dari Variabel(). |
lr | Faktor skala. Pasti skalar. |
lulusan | Gradien. |
momentum | momentum. Pasti skalar. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru ResourceApplyKerasMomentum
ResourceApplyKerasMomentum.Options statis publik 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. |
---|
ResourceApplyKerasMomentum.Options statis publik useNesterov (Boolean useNesterov)
Parameter
gunakanNesterov | Jika `True`, tensor yang diteruskan ke komputasi grad akan menjadi var + momentum * accum, sehingga pada akhirnya, var yang Anda dapatkan sebenarnya adalah var + momentum * accum. |
---|