ApplyMomentum.Options

classe estática pública ApplyMomentum.Options

Atributos opcionais para ApplyMomentum

Métodos Públicos

AplicarMomentum.Options
useLocking (booleano useLocking)
AplicarMomentum.Options
useNesterov (booleano useNesterov)

Métodos herdados

Métodos Públicos

public ApplyMomentum.Options useLocking (booleano useLocking)

Parâmetros
useLocking Se `True`, a atualização dos tensores var e accum será protegida por um bloqueio; caso contrário, o comportamento será indefinido, mas poderá apresentar menos contenção.

public ApplyMomentum.Options useNesterov (booleano useNesterov)

Parâmetros
usarNesterov Se `True`, o tensor passado para calcular grad será var - lr * momentum * accum, então, no final, o var que você obtém é na verdade var - lr * momentum * accum.