diff --git a/examples/visualize_optimizers.py b/examples/visualize_optimizers.py index 9ebe50ccc..e13f8e991 100644 --- 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))