Added Automagic v3

2026-06-07 12:06:42 -06:00
parent 82190b41e6
commit 1dc9a797cf
3 changed files with 378 additions and 29 deletions
--- a/toolkit/optimizer.py
+++ b/toolkit/optimizer.py
@@ -100,6 +100,9 @@ def get_optimizer(
    elif lower_type == 'automagic2':
        from toolkit.optimizers.automagic2 import Automagic2
        optimizer = Automagic2(params, lr=float(learning_rate), **optimizer_params)
+    elif lower_type == 'automagic3':
+        from toolkit.optimizers.automagic3 import Automagic3
+        optimizer = Automagic3(params, lr=float(learning_rate), **optimizer_params)
    else:
        raise ValueError(f'Unknown optimizer type {optimizer_type}')
    return optimizer