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