From b56d1bccd066b286d5a0fda51cdff3496d26481c Mon Sep 17 00:00:00 2001 From: Ahmed Khaled Date: Fri, 8 May 2026 16:25:24 -0700 Subject: [PATCH] internal PiperOrigin-RevId: 912748901 --- init2winit/optimizer_lib/optimizers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/init2winit/optimizer_lib/optimizers.py b/init2winit/optimizer_lib/optimizers.py index 0ef291a5..9db79ccc 100644 --- a/init2winit/optimizer_lib/optimizers.py +++ b/init2winit/optimizer_lib/optimizers.py @@ -517,7 +517,7 @@ def get_optimizer(hps, model=None, batch_axis_name=None): opt_init, opt_update = ( sharpness_aware_minimization.sharpness_aware_minimization( rho=hps.opt_hparams['rho'], - grad_clip=hps.get('grad_clip', None), + grad_clip=hps.opt_hparams.get('grad_clip', None), base_opt_init_fn=opt_init, base_opt_update_fn=opt_update, )