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