מחלקה סופית ציבורית LoadAllTPUEmbeddingParameters
הפעלה שטוענת פרמטרי אופטימיזציה לתוך הטבעת זיכרון.
הפעלה שטוענת פרמטרי אופטימיזציה לתוך הטבעת זיכרון. יש להקדים את הפעלת ConfigureTPUEmbeddingHost המגדירה את תצורת טבלת ההטמעה הנכונה. לדוגמה, אופציה זו משמשת להתקנת פרמטרים שנטענים מנקודת ביקורת לפני ביצוע לולאת אימון. עבור אדגרד, עזר1 צריך להיות המצברים. עבור SGD, כל ערכי העזר* צריכים להיות ריקים. עבור FTRL, auxiliary1 צריך להיות המצברים ו-auxiliary2 צריך להיות המונחים הליניאריים. עבור ADAM, auxiliary1 צריך להיות המומנטה ו-auxiliary2 צריך להיות המהירויות.
שיטות ציבוריות
| סטטי LoadAllTPUEmbeddingParameters | create ( scope scope, 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 חדשה. |
שיטות בירושה
שיטות ציבוריות
Public static LoadAllTPUEmbeddingParameters create ( scope scope, 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 | רשימה של טנזורים, אחד עבור כל טבלת הטבעה, המכילה את הערכים ההתחלתיים של פרמטר האופטימיזציה העזר השלישי לשימוש בהטמעת עדכוני לולאת אימון. הצורה של כל ערך מתעלמת (ולכן יכולה להיות ריקה) עבור הטבלאות שלאלגוריתמי האופטימיזציה שלהן אין שבעה פרמטרים עזר. |
| תצורה | פרוטו של TPUembeddingConfiguration המתאר את פרמטרי הטבלה הנטענים, מסודרים למחרוזת. |
| numShards | מספר הרסיסים שאליהם מחולקים טבלאות ההטבעה. |
| shardId | מזהה של רסיס עבור פעולה זו. |
החזרות
- מופע חדש של LoadAllTPUEmbeddingParameters