From 2acade3025b61d294c3fcceb1bf5193e6d984b44 Mon Sep 17 00:00:00 2001 From: kozistr Date: Sun, 14 Jul 2024 13:36:59 +0900 Subject: [PATCH] update: test_lomo_optimizer --- examples/visualize_optimizers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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))