diff options
| author | Max Ryabinin <mryabinin0@gmail.com> | 2022-07-01 17:16:10 +0300 |
|---|---|---|
| committer | Max Ryabinin <mryabinin0@gmail.com> | 2022-07-01 17:16:10 +0300 |
| commit | 8258b4364a21a4da2572cb644d0926080c3268da (patch) | |
| tree | 571e95bc327116fbaba08d14871fb0b224b8a65b /bitsandbytes/optim/rmsprop.py | |
| parent | 33efe4a09f459832e8beceba70add0695cc485e4 (diff) | |
Add a CPU-only build option
Diffstat (limited to 'bitsandbytes/optim/rmsprop.py')
| -rw-r--r-- | bitsandbytes/optim/rmsprop.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/bitsandbytes/optim/rmsprop.py b/bitsandbytes/optim/rmsprop.py index 7909d5d..0f1ffaa 100644 --- a/bitsandbytes/optim/rmsprop.py +++ b/bitsandbytes/optim/rmsprop.py @@ -31,6 +31,6 @@ class RMSprop32bit(Optimizer1State): if alpha == 0: raise NotImplementedError(f'RMSprop with alpha==0.0 is not supported!') if centered: - raise NotImplementError(f'Centered RMSprop is not supported!') + raise NotImplementedError(f'Centered RMSprop is not supported!') super(RMSprop32bit, self).__init__('rmsprop', params, lr, (alpha, momentum), eps, weight_decay, 32, args, min_8bit_size, percentile_clipping, block_wise) |