SparseApplyMomentum.Options

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.