classe statique publique SparseApplyMomentum.Options
Attributs facultatifs pour SparseApplyMomentum
Méthodes publiques
SparseApplyMomentum.Options | useLocking (booléen useLocking) |
SparseApplyMomentum.Options | useNesterov (booléen useNesterov) |
Méthodes héritées
Méthodes publiques
public SparseApplyMomentum.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 SparseApplyMomentum.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. |
---|