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

Fixed experiment opt name match

Browse Source
This commit is contained in:
Jaret Burkett
2026-08-01 17:43:36 -06:00
parent 6b95282097
commit a6f6b6b896
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/optimizer.py
View File

@@ -103,7 +103,7 @@ 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':
elif lower_type == 'automagicexperiment':
from toolkit.optimizers.automagicEXPERIMENT import AutomagicEXPERIMENT
optimizer = AutomagicEXPERIMENT(params, lr=float(learning_rate), **optimizer_params)
else: