ResourceApplyMomentum.Options

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.