Merge pull request #3424 from zh794390558/fix_opt

fix weight decay

Merge pull request #3424 from zh794390558/fix_opt
fix weight decay
a1745944 · Hui Zhang · GitHub · 5d10d6e8 · 2faa49a3 · a1745944
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

paddlespeech/s2t/training/optimizer/__init__.py paddlespeech/s2t/training/optimizer/__init__.py +1 -2

未找到文件。
--- 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: