כיתת גמר ציבורית ResourceApplyKerasMomentum
עדכן את '*var' לפי סכימת המומנטום.
הגדר use_nesterov = True אם אתה רוצה להשתמש במומנטום Nesterov.
accum = accum * מומנטום - lr * grad var += accum
כיתות מקוננות
מעמד | ResourceApplyKerasMomentum.Options | תכונות אופציונליות עבור ResourceApplyKerasMomentum |
קבועים
חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
static <T מרחיב את TType > ResourceApplyKerasMomentum | |
static ResourceApplyKerasMomentum.Options | useLocking (useLocking בוליאני) |
static ResourceApplyKerasMomentum.Options | useNesterov (בוליאני useNesterov) |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
ערך קבוע: "ResourceApplyKerasMomentum"
שיטות ציבוריות
public static ResourceApplyKerasMomentum create ( scope scope, Operand <?> var, Operand <?> accum, Operand <T> lr, Operand <T> grad, Operand <T> momentum, Options... options)
שיטת מפעל ליצירת מחלקה העוטפת פעולת ResourceApplyKerasMomentum חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
var | צריך להיות ממשתנה(). |
לצבור | צריך להיות ממשתנה(). |
lr | גורם קנה מידה. חייב להיות סקלר. |
גראד | השיפוע. |
תְנוּפָה | תְנוּפָה. חייב להיות סקלר. |
אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של ResourceApplyKerasMomentum
public static ResourceApplyKerasMomentum.Options useLocking (שימוש בוליאני)
פרמטרים
השתמש בנעילה | אם 'נכון', עדכון טנסור ה- var ו-acum יהיה מוגן על ידי מנעול; אחרת ההתנהגות אינה מוגדרת, אך עלולה להפגין פחות מחלוקת. |
---|
public static ResourceApplyKerasMomentum.Options useNesterov (בוליאני useNesterov)
פרמטרים
השתמש ב-Nesterov | אם 'נכון', הטנזור שיעבור לחישוב גראד יהיה var + מומנטום * accum, כך שבסופו של דבר, ה-var שאתה מקבל הוא למעשה var + momentum * accum. |
---|