update: test_lomo_optimizer

kozistr · Jul 14, 2024 · 2acade3 · 2acade3
1 parent c3b4309
commit 2acade3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/visualize_optimizers.py b/examples/visualize_optimizers.py
@@ -158,7 +158,7 @@ def main():
     ]
 
     for optimizer_name, optimizer in OPTIMIZERS.items():
-        if optimizer_name.lower() in {'alig', 'lomo', 'bsam', 'adammini'}:
+        if optimizer_name.lower() in {'alig', 'lomo', 'adalomo', 'bsam', 'adammini'}:
             continue
 
         optimizers.append((optimizer, -6, 0.2))