LoadAllTPUEmbeddingParameters

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