LoadTPUEmbeddingADAMParametersGradAccumDebug

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

טען פרמטרים של הטמעת ADAM עם תמיכה באגים.

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

כיתות מקוננות

מעמד LoadTPUEmbeddingADAMPparametersGradAccumDebug.Options מאפיינים אופציונליים עבור LoadTPUEmbeddingADAMParametersGradAccumDebug

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

סטטי LoadTPUEmbeddingADAMPparametersGradAccumDebug.Options
config (תצורת מחרוזת)
סטטי LoadTPUEmbeddingADAMPparametersGradAccumDebug
create ( scope scope, Operand <Float> פרמטרים, Operand <Float> מומנטה, Operand <Float> מהירויות, Operand <Float> gradientAccumulators, Long numShards, Long shardId, Options... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת LoadTPUEmbeddingADAMParametersGradAccumDebug חדשה.
סטטי LoadTPUEmbeddingADAMPparametersGradAccumDebug.Options
tableId (Long tableId)
סטטי LoadTPUEmbeddingADAMPparametersGradAccumDebug.Options
tableName (מחרוזת tableName)

שיטות בירושה

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

public static LoadTPUEmbeddingADAMPparametersGradAccumDebug.Options config (String config)

public static LoadTPUEmbeddingADAMParametersGradAccumDebug create ( היקף היקף , פרמטרים של Operand <Float>, מומנטה של ​​Operand <Float>, מהירויות Operand <Float>, GradientAccumulators של Operand , Long numShards, Long shardId, אפשרויות... אפשרויות)

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

פרמטרים
תְחוּם ההיקף הנוכחי
פרמטרים ערך הפרמטרים המשמשים באלגוריתם האופטימיזציה של ADAM.
מומנטה ערך המומנטה בשימוש באלגוריתם האופטימיזציה של ADAM.
מהירויות ערך המהירויות בשימוש באלגוריתם האופטימיזציה של ADAM.
gradientAccumulators ערך של gradient_accumulators בשימוש באלגוריתם האופטימיזציה של ADAM.
אפשרויות נושא ערכי תכונות אופציונליות
החזרות
  • מופע חדש של LoadTPUEmbeddingADAMParametersGradAccumDebug

public static LoadTPUEmbeddingADAMPparametersGradAccumDebug.Options tableId (Long tableId)

public static LoadTPUEmbeddingADAMPparametersGradAccumDebug.Options tableName (String tableName)