diff --git a/paddlespeech/s2t/training/optimizer/__init__.py b/paddlespeech/s2t/training/optimizer/__init__.py index 0f998ddeca552d7057f56bdc3575d3182d161cd5..90281e1ed8cd89474453d4f0becdaa97b8afe3ea 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: