SparseApplyMomentum.Options

clase estática pública SparseApplyMomentum.Options

Atributos opcionales para SparseApplyMomentum

Métodos públicos

SparseApplyMomentum.Opciones
useLocking (uso booleano Locking)
SparseApplyMomentum.Opciones
useNesterov (uso booleanoNesterov)

Métodos heredados

Métodos públicos

público SparseApplyMomentum.Options useLocking (uso booleano Locking)

Parámetros
utilizarBloqueo Si es "True", la actualización de los tensores var y accum estará protegida por un bloqueo; de lo contrario, el comportamiento no está definido, pero puede presentar menos contención.

público SparseApplyMomentum.Options useNesterov (uso booleanoNesterov)

Parámetros
utilizarNesterov Si es "Verdadero", el tensor pasado para calcular grad será var - lr * momento * accum, por lo que al final, el var que obtienes es en realidad var - lr * momento * accum.