forget about tests for now, will test live on local enwik8 training

8de29fc3 · Phil Wang · cb4c3c8c · 8de29fc3
Commit 8de29fc3 authored Mar 09, 2023 by Phil Wang
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 4 deletions

tests/test_optim.py tests/test_optim.py +0 -4

No files found.
--- a/tests/test_optim.py
+++ b/tests/test_optim.py
@@ -50,10 +50,6 @@ str2optimizers["rmsprop"] = (
    lambda pxx: torch.optim.RMSprop(pxx, 0.01, 0.9),
    lambda pxx: bnb.optim.RMSprop(pxx, 0.01, 0.9, block_wise=False),
 )
-str2optimizers["rmsprop"] = (
-    lambda pxx: torch.optim.RMSprop(pxx, 0.01, 0.9),
-    lambda pxx: bnb.optim.RMSprop(pxx, 0.01, 0.9, block_wise=False),
-)
 str2optimizers["adam8bit"] = (
    torch.optim.Adam,
    lambda pxx: bnb.optim.Adam8bit(pxx, block_wise=False),