tensorflow::ops::ResourceApplyAdam
#include <training_ops.h>อัปเดต '*var' ตามอัลกอริทึมของ Adam
สรุป
$$lr_t := {learning_rate} * {1 - beta_2^t} / (1 - beta_1^t)$$ $$m_t := beta_1 * m_{t-1} + (1 - beta_1) * g$$ $$v_t := beta_2 * v_{t-1} + (1 - beta_2) * g * g$$ $$variable := variable - lr_t * m_t / ({v_t} + )$$
ข้อโต้แย้ง:
- ขอบเขต: วัตถุ ขอบเขต
- var: ควรมาจาก Variable()
- m: ควรมาจากตัวแปร()
- v: ควรมาจากตัวแปร (Variable())
- beta1_power: ต้องเป็นค่าสเกลาร์
- beta2_power: ต้องเป็นค่าสเกลาร์
- lr: ตัวคูณปรับขนาด ต้องเป็นค่าคงที่
- เบต้า1: ตัวประกอบโมเมนตัม ต้องเป็นค่าสเกลาร์
- เบต้า2: ตัวประกอบโมเมนตัม ต้องเป็นค่าสเกลาร์
- เอปซิลอน: เทอมของสัน ต้องเป็นค่าสเกลาร์
- grad: ค่าความชัน
คุณลักษณะเสริม (ดูที่ Attrs ):
- use_locking: หากตั้งค่า
Trueการอัปเดตเทนเซอร์ var, m และ v จะได้รับการป้องกันด้วยการล็อก มิเช่นนั้นพฤติกรรมจะไม่แน่นอน แต่การแย่งชิงทรัพยากรอาจน้อยลง - use_nesterov: ถ้า
Trueจะใช้การอัปเดตแบบ nesterov
การส่งคืนสินค้า:
-
Operationที่สร้างขึ้น
ตัวสร้างและตัวทำลาย | |
|---|---|
ResourceApplyAdam (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input m, :: tensorflow::Input v, :: tensorflow::Input beta1_power, :: tensorflow::Input beta2_power, :: tensorflow::Input lr, :: tensorflow::Input beta1, :: tensorflow::Input beta2, :: tensorflow::Input epsilon, :: tensorflow::Input grad) | |
ResourceApplyAdam (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input m, :: tensorflow::Input v, :: tensorflow::Input beta1_power, :: tensorflow::Input beta2_power, :: tensorflow::Input lr, :: tensorflow::Input beta1, :: tensorflow::Input beta2, :: tensorflow::Input epsilon, :: tensorflow::Input grad, const ResourceApplyAdam::Attrs & attrs) |
คุณลักษณะสาธารณะ | |
|---|---|
operation | |
หน้าที่สาธารณะ | |
|---|---|
operator::tensorflow::Operation () const | |
ฟังก์ชันคงที่สาธารณะ | |
|---|---|
UseLocking (bool x) | |
UseNesterov (bool x) | |
โครงสร้าง | |
|---|---|
| tensorflow::ops::ResourceApplyAdam::Attrs | ตัวเลือกในการตั้งค่าแอตทริบิวต์สำหรับ ResourceApplyAdam |
คุณลักษณะสาธารณะ
การดำเนินการ
Operation operation
หน้าที่สาธารณะ
ทรัพยากรใช้ Adam
ResourceApplyAdam( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input m, ::tensorflow::Input v, ::tensorflow::Input beta1_power, ::tensorflow::Input beta2_power, ::tensorflow::Input lr, ::tensorflow::Input beta1, ::tensorflow::Input beta2, ::tensorflow::Input epsilon, ::tensorflow::Input grad )
ทรัพยากรใช้ Adam
ResourceApplyAdam( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input m, ::tensorflow::Input v, ::tensorflow::Input beta1_power, ::tensorflow::Input beta2_power, ::tensorflow::Input lr, ::tensorflow::Input beta1, ::tensorflow::Input beta2, ::tensorflow::Input epsilon, ::tensorflow::Input grad, const ResourceApplyAdam::Attrs & attrs )
ตัวดำเนินการ::เทนเซอร์โฟลว์::การดำเนินการ
operator::tensorflow::Operation() const
ฟังก์ชันคงที่สาธารณะ
ใช้การล็อก
Attrs UseLocking( bool x )
UseNesterov
Attrs UseNesterov( bool x )