bitsandbytes/pr_2032/en/reference/optim/rmsprop.md

RMSprop

RMSprop is an adaptive learning rate optimizer that is very similar to Adagrad. RMSprop stores a weighted average of the squared past gradients for each parameter and uses it to scale their learning rate. This allows the learning rate to be automatically lower or higher depending on the magnitude of the gradient, and it prevents the learning rate from diminishing.

RMSprop[[api-class]][[bitsandbytes.optim.RMSprop]]

bitsandbytes.optim.RMSprop[[bitsandbytes.optim.RMSprop]]

bitsandbytes.optim.RMSprop(params, lr = 0.01, alpha = 0.99, eps = 1e-08, weight_decay = 0, momentum = 0, centered = False, optim_bits = 32, args = None, min_8bit_size = 4096)

Source

RMSprop8bit[[bitsandbytes.optim.RMSprop8bit]]

bitsandbytes.optim.RMSprop8bit[[bitsandbytes.optim.RMSprop8bit]]

bitsandbytes.optim.RMSprop8bit(params, lr = 0.01, alpha = 0.99, eps = 1e-08, weight_decay = 0, momentum = 0, centered = False, args = None, min_8bit_size = 4096)

Source

RMSprop32bit[[bitsandbytes.optim.RMSprop32bit]]

bitsandbytes.optim.RMSprop32bit[[bitsandbytes.optim.RMSprop32bit]]

bitsandbytes.optim.RMSprop32bit(params, lr = 0.01, alpha = 0.99, eps = 1e-08, weight_decay = 0, momentum = 0, centered = False, args = None, min_8bit_size = 4096)

Source