โหลดพารามิเตอร์การฝังโมเมนตัมพร้อมการสนับสนุนการแก้ไขข้อบกพร่อง
ปฏิบัติการที่โหลดพารามิเตอร์การปรับให้เหมาะสมลงใน HBM เพื่อการฝัง ต้องนำหน้าด้วยตัวเลือก ConfigureTPUEmbeddingHost ที่ตั้งค่าการกำหนดค่าตารางฝังที่ถูกต้อง ตัวอย่างเช่น op นี้ใช้เพื่อติดตั้งพารามิเตอร์ที่โหลดจากจุดตรวจสอบก่อนที่จะดำเนินการลูปการฝึก
คลาสที่ซ้อนกัน
ระดับ | โหลด TPU การฝังโมเมนตัมพารามิเตอร์ GradAccum Debug.Options | คุณลักษณะเพิ่มเติมสำหรับ LoadTPUEmbeddingMomentumParametersGradAccumDebug |
ค่าคงที่
สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options แบบคงที่ | config (กำหนดค่าสตริง) |
LoadTPU การฝังโมเมนตัมพารามิเตอร์ GradAccumDebug แบบคงที่ | สร้าง (ขอบเขต ขอบเขต ตัวดำเนินการ < TFloat32 > พารามิเตอร์ ตัวดำเนินการ < TFloat32 > โมเมนต้า ตัวดำเนินการ < TFloat32 > การไล่ระดับสีAccumulators, Long numShards, Long shardId, ตัวเลือก... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ LoadTPUEmbeddingMomentumParametersGradAccumDebug ใหม่ |
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options แบบคงที่ | tableId (รหัสตารางแบบยาว) |
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options แบบคงที่ | tableName (ชื่อตารางสตริง) |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
วิธีการสาธารณะ
LoadTPUEmbeddingMomentumParametersGradAccumDebug สาธารณะ สร้าง (ขอบเขต ขอบเขต , ตัวดำเนินการ < TFloat32 > พารามิเตอร์, ตัวดำเนินการ < TFloat32 > โมเมนต้า, ตัวดำเนินการ < TFloat32 > การไล่ระดับสีสะสม, ยาว numShards, ยาว shardId, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ LoadTPUEmbeddingMomentumParametersGradAccumDebug ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
พารามิเตอร์ | ค่าของพารามิเตอร์ที่ใช้ในอัลกอริธึมการปรับให้เหมาะสมของโมเมนตัม |
โมเมนตัม | มูลค่าของโมเมนต้าที่ใช้ในอัลกอริธึมการหาค่าเหมาะที่สุดสำหรับโมเมนตัม |
การไล่ระดับสีสะสม | ค่าของ gradient_accumulators ที่ใช้ในอัลกอริธึมการหาค่าเหมาะที่สุดสำหรับโมเมนตัม |
ตัวเลือก | มีค่าแอตทริบิวต์ทางเลือก |
การส่งคืน
- อินสแตนซ์ใหม่ของ LoadTPUEmbeddingMomentumParametersGradAccumDebug