diff --git a/toolkit/optimizer.py b/toolkit/optimizer.py index c6a3298..f1c86d8 100644 --- a/toolkit/optimizer.py +++ b/toolkit/optimizer.py @@ -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: