Fixed adafactor eps

This commit is contained in:
Jaret Burkett committed 2025-10-26 05:47:25 -06:00
1 parent 80418209b8
commit 8c12977891
1 file changed
+1 -1
+1 -1
View File
@@ -93,7 +93,7 @@ def get_optimizer(
optimizer_params['scale_parameter'] = False
if 'warmup_init' not in optimizer_params:
optimizer_params['warmup_init'] = False
optimizer = Adafactor(params, lr=float(learning_rate), eps=1e-6, **optimizer_params)
optimizer = Adafactor(params, lr=float(learning_rate), **optimizer_params)
elif lower_type == 'automagic':
from toolkit.optimizers.automagic import Automagic
optimizer = Automagic(params, lr=float(learning_rate), **optimizer_params)