คลาสสุดท้ายสาธารณะ LoadAllTPUEmbeddingParameters
op ที่โหลดพารามิเตอร์การเพิ่มประสิทธิภาพลงในหน่วยความจำที่ฝัง
op ที่โหลดพารามิเตอร์การเพิ่มประสิทธิภาพลงในหน่วยความจำที่ฝัง ต้องนำหน้าด้วย ConfigureTPUEmbeddingHost op ที่ตั้งค่าการกำหนดค่าตารางการฝังที่ถูกต้อง ตัวอย่างเช่น op นี้ใช้เพื่อติดตั้งพารามิเตอร์ที่โหลดจากจุดตรวจสอบก่อนที่จะดำเนินการวนรอบการฝึก สำหรับ Adagrad ตัวช่วย 1 ควรเป็นตัวสะสม สำหรับ SGD ค่าเสริม* ทั้งหมดควรว่างเปล่า สำหรับ FTRL ตัวช่วย1 ควรเป็นตัวสะสม และตัวช่วย2 ควรเป็นเงื่อนไขเชิงเส้น สำหรับ ADAM ตัวช่วย1 ควรเป็นโมเมนต์และตัวช่วย2 ควรเป็นความเร็ว
วิธีการสาธารณะ
| สแตติก LoadAllTPUEmbeddingParameters | สร้าง ( ขอบเขต ขอบเขต Iterable< Operand <Float>> พารามิเตอร์ Iterable< Operand <Float>> auxiliary1, Iterable< Operand <Float>> auxiliary2, Iterable< Operand <Float>> auxiliary3, Iterable< Operand <Float>> auxiliary4, Iterable< Operand <Float>> auxiliary5, Iterable< Operand <Float>> auxiliary6, Iterable< Operand <Float>> auxiliary7, String config, numShards แบบยาว, Long shardId) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ LoadAllTPUEmbeddingParameters ใหม่ |
วิธีการสืบทอด
วิธีการสาธารณะ
LoadAllTPUEmbeddingParameters สาธารณะแบบคงที่ สร้าง ( ขอบเขต ขอบเขต Iterable< Operand <Float>> พารามิเตอร์ Iterable< Operand <Float>> auxiliary1, Iterable< Operand <Float>> auxiliary2, Iterable< Operand <Float>> auxiliary3, Iterable< Operand <Float> > auxiliary4, Iterable< Operand <Float>> auxiliary5, Iterable< Operand <Float>> auxiliary6, Iterable< Operand <Float>> auxiliary7, String config, Long numShards, Long shardId)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ LoadAllTPUEmbeddingParameters ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|---|
| พารามิเตอร์ | รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางการฝัง ที่มีพารามิเตอร์ตารางการฝังเริ่มต้นเพื่อใช้ในการค้นหาการฝัง |
| เสริม1 | รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางการฝัง ที่มีค่าเริ่มต้นของพารามิเตอร์การปรับให้เหมาะสมเสริมแรกที่จะใช้ในการฝังการอัปเดตลูปการฝึก รูปร่างของแต่ละรายการจะถูกละเว้น (และสามารถว่างเปล่าได้) สำหรับตารางเหล่านั้นซึ่งอัลกอริธึมการปรับให้เหมาะสมไม่มีพารามิเตอร์เสริมอย่างน้อยหนึ่งตัว |
| เสริม2 | รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางการฝัง ที่มีค่าเริ่มต้นของพารามิเตอร์การปรับให้เหมาะสมรองที่สองเพื่อใช้ในการฝังการอัปเดตลูปการฝึก รูปร่างของแต่ละรายการจะถูกละเว้น (และสามารถว่างเปล่าได้) สำหรับตารางเหล่านั้นซึ่งอัลกอริธึมการปรับให้เหมาะสมไม่มีตัวช่วยอย่างน้อยสองตัว |
| เสริม3 | รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางการฝัง ที่มีค่าเริ่มต้นของพารามิเตอร์การปรับให้เหมาะสมเสริมที่สามเพื่อใช้ในการฝังการอัปเดตลูปการฝึก รูปร่างของแต่ละรายการจะถูกละเว้น (และสามารถว่างเปล่าได้) สำหรับตารางเหล่านั้นซึ่งอัลกอริธึมการปรับให้เหมาะสมไม่มีพารามิเตอร์เสริมสามตัว |
| เสริม4 | รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางการฝัง ที่มีค่าเริ่มต้นของพารามิเตอร์การปรับให้เหมาะสมรองที่สองเพื่อใช้ในการฝังการอัปเดตลูปการฝึก รูปร่างของแต่ละรายการจะถูกละเว้น (และสามารถว่างเปล่าได้) สำหรับตารางเหล่านั้นซึ่งอัลกอริธึมการปรับให้เหมาะสมไม่มีตัวช่วยอย่างน้อยสี่ตัว |
| เสริม5 | รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางการฝัง ที่มีค่าเริ่มต้นของพารามิเตอร์การปรับให้เหมาะสมเสริมที่สามเพื่อใช้ในการฝังการอัปเดตลูปการฝึก รูปร่างของแต่ละรายการจะถูกละเว้น (และสามารถว่างเปล่าได้) สำหรับตารางเหล่านั้นซึ่งอัลกอริธึมการปรับให้เหมาะสมไม่มีพารามิเตอร์เสริมห้าตัว |
| เสริม6 | รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางการฝัง ที่มีค่าเริ่มต้นของพารามิเตอร์การปรับให้เหมาะสมรองที่สองเพื่อใช้ในการฝังการอัปเดตลูปการฝึก รูปร่างของแต่ละรายการจะถูกละเว้น (และสามารถว่างเปล่าได้) สำหรับตารางเหล่านั้นซึ่งอัลกอริธึมการปรับให้เหมาะสมไม่มีตัวช่วยอย่างน้อยหกตัว |
| เสริม7 | รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางการฝัง ที่มีค่าเริ่มต้นของพารามิเตอร์การปรับให้เหมาะสมเสริมที่สามเพื่อใช้ในการฝังการอัปเดตลูปการฝึก รูปร่างของแต่ละรายการจะถูกละเว้น (และสามารถว่างเปล่าได้) สำหรับตารางเหล่านั้นซึ่งอัลกอริธึมการปรับให้เหมาะสมไม่มีพารามิเตอร์เสริม sevan |
| config | โปรโตไทป์ TPUEmbeddingConfiguration ที่อธิบายพารามิเตอร์ตารางที่กำลังโหลด จัดลำดับเป็นสตริง |
| numShards | จำนวนชาร์ดที่แบ่งตารางการฝัง |
| shardId | ตัวระบุชาร์ดสำหรับการดำเนินการนี้ |
คืนสินค้า
- อินสแตนซ์ใหม่ของ LoadAllTPUEmbeddingParameters