การดำเนินการที่โหลดพารามิเตอร์การปรับให้เหมาะสมลงในหน่วยความจำแบบฝัง
การดำเนินการที่โหลดพารามิเตอร์การปรับให้เหมาะสมลงในหน่วยความจำแบบฝัง ต้องนำหน้าด้วยตัวเลือก 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