evan/ai-toolkit
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Automagic V4

Browse Source
This commit is contained in:
Jaret Burkett committed 2026-10-23 14:00:27 -06:00
1 parent ea47c120aa
commit 534d462ff6
3 files changed
+263
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

toolkit/optimizer.py
+3
View File
@@ -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 == 'automagic4':
from toolkit.optimizers.automagic4 import Automagic4
optimizer = Automagic4(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)