RetrieveAllTPUEmbeddingParameters

מחלקה סופית ציבורית RetrieveAllTPUEmbeddingParameters

אופ שמחזיר את פרמטרי האופטימיזציה מהטמעה לזיכרון המארח.

אופ שמחזיר את פרמטרי האופטימיזציה מהטמעה לזיכרון המארח. יש להקדים את הפעלת ConfigureTPUEmbeddingHost המגדירה את תצורת טבלת ההטמעה הנכונה. לדוגמה, אופציה זו משמשת לאחזור פרמטרים מעודכנים לפני שמירת נקודת ביקורת. עבור Adagrad, auxiliary1 יכיל את המצברים לאחר הפעלת הפעולה הזו. עבור SGD, כל ערכי העזר* יהיו ריקים (0x0 טנסורים עבור הטבלה הזו). עבור FTRL, auxiliary1 יכיל את המצברים ו-auxiliary2 יכיל את המונחים הליניאריים. עבור ADAM, auxiliary1 יכיל את המומנטה ו-auxiliary2 יכיל את המהירויות.

שיטות ציבוריות

רשימה< פלט <צף>>
עזר1 ()
רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את פרמטר האופטימיזציה העזר הראשון המאוחסן.
רשימה< פלט <צף>>
עזר2 ()
רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את פרמטר האופטימיזציה העזר השני המאוחסן.
רשימה< פלט <צף>>
עזר3 ()
רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את פרמטר האופטימיזציה העזר השלישי המאוחסן.
רשימה< פלט <צף>>
עזר4 ()
רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את פרמטר האופטימיזציה העזר הרביעי המאוחסן.
רשימה< פלט <צף>>
עזר5 ()
רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את פרמטר האופטימיזציה העזר החמישי המאוחסן.
רשימה< פלט <צף>>
עזר6 ()
רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את ששת פרמטר האופטימיזציה העזר המאוחסן.
רשימה< פלט <צף>>
עזר7 ()
רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את פרמטר האופטימיזציה העזר השביעי המאוחסן.
static RetrieveAllTPUEmbeddingParameters
ליצור (טווח היקף , Long NumTables, String config, Long numShards, Long shardId)
שיטת מפעל ליצירת מחלקה העוטפת פעולת RetrieveAllTPUEmbeddingParameters חדשה.
רשימה< פלט <צף>>
פרמטרים ()
רשימה של טנזורים, אחד לכל טבלת הטבעה, המכילה את הפרמטרים המאוחסנים של טבלת ההטמעה.

שיטות בירושה

שיטות ציבוריות

רשימה ציבורית< פלט <צף>> auxiliary1 ()

רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את פרמטר האופטימיזציה העזר הראשון המאוחסן. אלמנטים קיימים ברשימה, אך הם בעלי גודל אפס, עבור פרמטרי אופטימיזציה שאינם בשימוש (בהתבסס על האלגוריתם בשימוש עבור כל טבלה).

רשימה ציבורית< פלט <Float>> auxiliary2 ()

רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את פרמטר האופטימיזציה העזר השני המאוחסן. אלמנטים קיימים ברשימה, אך הם בעלי גודל אפס, עבור פרמטרי אופטימיזציה שאינם בשימוש (בהתבסס על האלגוריתם בשימוש עבור כל טבלה).

רשימה ציבורית< פלט <Float>> auxiliary3 ()

רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את פרמטר האופטימיזציה העזר השלישי המאוחסן. אלמנטים קיימים ברשימה, אך הם בעלי גודל אפס, עבור פרמטרי אופטימיזציה שאינם בשימוש (בהתבסס על האלגוריתם בשימוש עבור כל טבלה).

רשימה ציבורית< פלט <Float>> auxiliary4 ()

רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את פרמטר האופטימיזציה העזר הרביעי המאוחסן. אלמנטים קיימים ברשימה, אך הם בעלי גודל אפס, עבור פרמטרי אופטימיזציה שאינם בשימוש (בהתבסס על האלגוריתם בשימוש עבור כל טבלה).

רשימה ציבורית< פלט <Float>> auxiliary5 ()

רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את פרמטר האופטימיזציה העזר החמישי המאוחסן. אלמנטים קיימים ברשימה, אך הם בעלי גודל אפס, עבור פרמטרי אופטימיזציה שאינם בשימוש (בהתבסס על האלגוריתם בשימוש עבור כל טבלה).

רשימה ציבורית< פלט <Float>> auxiliary6 ()

רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את ששת פרמטר האופטימיזציה העזר המאוחסן. אלמנטים קיימים ברשימה, אך הם בעלי גודל אפס, עבור פרמטרי אופטימיזציה שאינם בשימוש (בהתבסס על האלגוריתם בשימוש עבור כל טבלה).

רשימה ציבורית< פלט <צף>> עזר7 ()

רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את פרמטר האופטימיזציה העזר השביעי המאוחסן. אלמנטים קיימים ברשימה, אך הם בעלי גודל אפס, עבור פרמטרי אופטימיזציה שאינם בשימוש (בהתבסס על האלגוריתם בשימוש עבור כל טבלה).

public static RetrieveAllTPUEmbeddingParameters create ( היקף היקף , Long NumTables, String config, Long numShards, Long shardId)

שיטת מפעל ליצירת מחלקה העוטפת פעולת RetrieveAllTPUEmbeddingParameters חדשה.

פרמטרים
תְחוּם ההיקף הנוכחי
מספר טבלאות מספר שולחנות הטבעה.
תצורה פרוטו של TPUembeddingConfiguration המתאר את פרמטרי הטבלה הנטענים, מסודרים למחרוזת.
numShards מספר הרסיסים שאליהם מחולקים טבלאות ההטבעה.
shardId מזהה של רסיס עבור פעולה זו.
החזרות
  • מופע חדש של RetrieveAllTPUEmbeddingParameters

רשימה ציבורית< פלט <צף>> פרמטרים ()

רשימה של טנזורים, אחד לכל טבלת הטבעה, המכילה את הפרמטרים המאוחסנים של טבלת ההטמעה.