ประชาชน ResourceApplyKerasMomentum ชั้นสุดท้าย
อัปเดต '*var' ตามรูปแบบโมเมนตัม
ตั้งค่า use_nesterov = True หากคุณต้องการใช้โมเมนตัม Nesterov
accum = accum * โมเมนตัม - lr * grad var += accum
คลาสที่ซ้อนกัน
ระดับ | ทรัพยากรApplyKerasMomentum.Options | คุณลักษณะที่จำเป็นสำหรับการ ResourceApplyKerasMomentum |
วิธีการสาธารณะ
คง <T> ResourceApplyKerasMomentum | |
คง ResourceApplyKerasMomentum.Options | useLocking (บูลีน useLocking) |
คง ResourceApplyKerasMomentum.Options | useNesterov (บูลีน useNesterov) |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง ResourceApplyKerasMomentum สร้าง ( ขอบเขต ขอบเขต Operand <?> var, Operand <?> accum, Operand <T> LR, Operand <T> จบ Operand <T> โมเมนตัม ตัวเลือก ... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ ResourceApplyKerasMomentum ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
var | ควรมาจาก Variable() |
สะสม | ควรมาจาก Variable() |
lr | ปัจจัยการปรับขนาด ต้องเป็นสเกลาร์ |
บัณฑิต | การไล่ระดับ |
โมเมนตัม | โมเมนตัม. ต้องเป็นสเกลาร์ |
ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ ResourceApplyKerasMomentum
สาธารณะคง ResourceApplyKerasMomentum.Options useLocking (บูลีน useLocking)
พารามิเตอร์
ใช้ล็อค | หากเป็น "จริง" การอัพเดทเทนเซอร์ var และ accum จะได้รับการป้องกันโดยการล็อค มิฉะนั้นพฤติกรรมจะไม่ถูกกำหนด แต่อาจแสดงความขัดแย้งน้อยลง |
---|
สาธารณะคง ResourceApplyKerasMomentum.Options useNesterov (บูลีน useNesterov)
พารามิเตอร์
ใช้ Nesterov | หากเป็น "จริง" เทนเซอร์ที่ส่งผ่านไปยังการคำนวณ grad จะเป็น var + โมเมนตัม * accum ดังนั้นในท้ายที่สุด var ที่คุณได้รับก็คือ var + โมเมนตัม * accum |
---|