Add code for automagic experiment

2026-08-01 10:04:46 -06:00
parent 497014bf5d
commit c68e58083f
3 changed files with 730 additions and 9 deletions
--- a/toolkit/optimizer.py
+++ b/toolkit/optimizer.py
@@ -103,6 +103,9 @@ def get_optimizer(
    elif lower_type == 'automagic3':
        from toolkit.optimizers.automagic3 import Automagic3
        optimizer = Automagic3(params, lr=float(learning_rate), **optimizer_params)
+    elif lower_type == 'automagicEXPERIMENT':
+        from toolkit.optimizers.automagicEXPERIMENT import AutomagicEXPERIMENT
+        optimizer = AutomagicEXPERIMENT(params, lr=float(learning_rate), **optimizer_params)
    else:
        raise ValueError(f'Unknown optimizer type {optimizer_type}')
    return optimizer