update: test_lomo_optimizer

kozistr · kozistr · commit 2acade3025b6 · 2024-07-14T13:36:59.000+09:00
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))

Original file line number	Diff line number	Diff line change
`@@ -158,7 +158,7 @@ def main():`
`158`	`158`	`]`
`159`	`159`
`160`	`160`	`for optimizer_name, optimizer in OPTIMIZERS.items():`
`161`		`- if optimizer_name.lower() in {'alig', 'lomo', 'bsam', 'adammini'}:`
	`161`	`+ if optimizer_name.lower() in {'alig', 'lomo', 'adalomo', 'bsam', 'adammini'}:`
`162`	`162`	`continue`
`163`	`163`
`164`	`164`	`optimizers.append((optimizer, -6, 0.2))`