From 2faa49a39fcd810a2c896f61006c6e9958a5e85c Mon Sep 17 00:00:00 2001 From: Hui Zhang Date: Tue, 25 Jul 2023 02:38:36 +0000 Subject: [PATCH] fix weight decay --- paddlespeech/s2t/training/optimizer/__init__.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/paddlespeech/s2t/training/optimizer/__init__.py b/paddlespeech/s2t/training/optimizer/__init__.py index 0f998dde..90281e1e 100644 --- a/paddlespeech/s2t/training/optimizer/__init__.py +++ b/paddlespeech/s2t/training/optimizer/__init__.py @@ -102,8 +102,7 @@ class OptimizerFactory(): grad_clip = paddle.nn.ClipGradByGlobalNorm( args['grad_clip']) if "grad_clip" in args else None - weight_decay = L2Decay( - args['weight_decay']) if "weight_decay" in args else None + weight_decay = args.get("weight_decay", None) if weight_decay: logger.info(f'') if grad_clip: