ApplyMomentum.Options

classe statique publique ApplyMomentum.Options

Attributs facultatifs pour ApplyMomentum

Méthodes publiques

AppliquerMomentum.Options
useLocking (booléen useLocking)
AppliquerMomentum.Options
useNesterov (booléen useNesterov)

Méthodes héritées

Méthodes publiques

public ApplyMomentum.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 ApplyMomentum.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.