מחלקה סופית ציבורית LoadTPUEbeddingADAMPparametersGradAccumDebug
טען פרמטרים של הטמעת ADAM עם תמיכה באגים.
הפעלה שטוענת פרמטרי אופטימיזציה לתוך HBM להטמעה. יש להקדים את הפעלת ConfigureTPUEmbeddingHost המגדירה את תצורת טבלת ההטמעה הנכונה. לדוגמה, אופציה זו משמשת להתקנת פרמטרים שנטענים מנקודת ביקורת לפני ביצוע לולאת אימון.
כיתות מקוננות
מעמד | LoadTPUEmbeddingADAMPparametersGradAccumDebug.Options | מאפיינים אופציונליים עבור LoadTPUEmbeddingADAMParametersGradAccumDebug |
קבועים
חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
סטטי LoadTPUEmbeddingADAMPparametersGradAccumDebug.Options | config (תצורת מחרוזת) |
סטטי LoadTPUEmbeddingADAMPparametersGradAccumDebug | |
סטטי LoadTPUEmbeddingADAMPparametersGradAccumDebug.Options | tableId (Long tableId) |
סטטי LoadTPUEmbeddingADAMPparametersGradAccumDebug.Options | tableName (מחרוזת tableName) |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
ערך קבוע: "LoadTPUEmbeddingADAMParametersGradAccumDebug"
שיטות ציבוריות
Public static LoadTPUEmbeddingADAMParametersGradAccumDebug create ( scope scope, Operand < TFloat32 > parameters, Operand < TFloat32 > momenta, Operand < TFloat32 > velocities, Operand < TFloat32 > gradientAccumulators, Long numShards, Long shardId)
שיטת מפעל ליצירת מחלקה העוטפת פעולת LoadTPUEmbeddingADAMParametersGradAccumDebug חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
פרמטרים | ערך הפרמטרים המשמשים באלגוריתם האופטימיזציה של ADAM. |
מומנטה | ערך המומנטה בשימוש באלגוריתם האופטימיזציה של ADAM. |
מהירויות | ערך המהירויות בשימוש באלגוריתם האופטימיזציה של ADAM. |
gradientAccumulators | ערך של gradient_accumulators בשימוש באלגוריתם האופטימיזציה של ADAM. |
אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של LoadTPUEmbeddingADAMParametersGradAccumDebug