diff options
| author | Tim Dettmers <TimDettmers@users.noreply.github.com> | 2022-07-18 09:51:37 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-07-18 09:51:37 -0700 |
| commit | 4cd7ea62b2f51c68aacde2f62e7141765e476111 (patch) | |
| tree | 548b2e77c62acd152330e898a6e17ea949a156d1 /bitsandbytes/optim/rmsprop.py | |
| parent | 3418cd390e952a7752fb6b2544c25e25af7c0371 (diff) | |
| parent | fd750cd2370b3b12e216a9148b23aaae63a80989 (diff) | |
Merge pull request #3 from TimDettmers/cpuonly
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) |