โหลดพารามิเตอร์การฝัง ADAM พร้อมการสนับสนุนการแก้ไขข้อบกพร่อง
ปฏิบัติการที่โหลดพารามิเตอร์การปรับให้เหมาะสมลงใน HBM เพื่อการฝัง ต้องนำหน้าด้วยตัวเลือก ConfigureTPUEmbeddingHost ที่ตั้งค่าการกำหนดค่าตารางฝังที่ถูกต้อง ตัวอย่างเช่น op นี้ใช้เพื่อติดตั้งพารามิเตอร์ที่โหลดจากจุดตรวจสอบก่อนที่จะดำเนินการลูปการฝึก
คลาสที่ซ้อนกัน
ระดับ | โหลดTPUEmbeddingADAMParametersGradAccumDebug.Options | แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingADAMParametersGradAccumDebug |
วิธีการสาธารณะ
LoadTPUEmbeddingADAMparametersGradAccumDebug.Options แบบคงที่ | config (กำหนดค่าสตริง) |
LoadTPUEmbeddingADAMparametersGradAccumDebug แบบคงที่ | สร้าง ( ขอบเขต ขอบเขต , พารามิเตอร์ Operand <Float>, โมเมนตัมของ Operand <Float>, ความเร็ว ของ Operand <Float>, Operand <Float> การไล่ระดับสี Accumulators, Long numShards, Long shardId, ตัวเลือก... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ LoadTPUEmbeddingADAMparametersGradAccumDebug ใหม่ |
LoadTPUEmbeddingADAMparametersGradAccumDebug.Options แบบคงที่ | tableId (รหัสตารางแบบยาว) |
LoadTPUEmbeddingADAMparametersGradAccumDebug.Options แบบคงที่ | tableName (ชื่อตารางสตริง) |
วิธีการสืบทอด
วิธีการสาธารณะ
LoadTPUEmbedding ADAMParametersGradAccumDebug สาธารณะแบบคงที่ ( ขอบเขต ขอบเขต , พารามิเตอร์ Operand <Float>, Operand <Float> โมเมนตา, Operand <Float> ความเร็ว, Operand <Float> การไล่ระดับสี Accumulators, Long numShards, Long shardId, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ LoadTPUEmbeddingADAMparametersGradAccumDebug ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
พารามิเตอร์ | ค่าพารามิเตอร์ที่ใช้ในอัลกอริธึมการปรับให้เหมาะสมของ ADAM |
โมเมนตัม | มูลค่าโมเมนต้าที่ใช้ในอัลกอริธึมการปรับให้เหมาะสมของ ADAM |
ความเร็ว | ค่าความเร็วที่ใช้ในอัลกอริธึมการหาค่าเหมาะที่สุด ADAM |
การไล่ระดับสีสะสม | ค่าของ gradient_accumulators ที่ใช้ในอัลกอริธึมการปรับให้เหมาะสมของ ADAM |
ตัวเลือก | มีค่าแอตทริบิวต์ทางเลือก |
การส่งคืน
- อินสแตนซ์ใหม่ของ LoadTPUEmbeddingADAMparametersGradAccumDebug