fix weight decay

12 months ago · 2faa49a39f
parent 5d10d6e884
commit 2faa49a39f
1 changed files with 1 additions and 2 deletions
--- 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(
+        weight_decay = args.get("weight_decay", None)
            args['weight_decay']) if "weight_decay" in args else None
        if weight_decay:
            logger.info(f'<WeightDecay - {weight_decay}>')
        if grad_clip: