classe statique publique ResourceApplyMomentum.Options
Attributs facultatifs pour ResourceApplyMomentum
Méthodes publiques
ResourceApplyMomentum.Options | useLocking (booléen useLocking) |
ResourceApplyMomentum.Options | useNesterov (booléen useNesterov) |
Méthodes héritées
Méthodes publiques
public ResourceApplyMomentum.Options useLocking (booléen useLocking)
Paramètres
utiliserVerrouillage | Si `True`, la mise à jour des tenseurs var et cumul sera protégée par un verrou ; sinon, le comportement n'est pas défini, mais peut présenter moins de conflits. |
---|
public ResourceApplyMomentum.Options useNesterov (booléen useNesterov)
Paramètres
utiliser Nesterov | Si « Vrai », le tenseur passé au calcul de la graduation sera var - lr * momentum * accum, donc à la fin, la var que vous obtenez est en fait var - lr * momentum * accum. |
---|