rmsprop.mdx 571 Bytes
Newer Older
Steven Liu's avatar
Steven Liu committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
# 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]]

[[autodoc]] bitsandbytes.optim.RMSprop

## RMSprop8bit

[[autodoc]] bitsandbytes.optim.RMSprop8bit

## RMSprop32bit

[[autodoc]] bitsandbytes.optim.RMSprop32bit