From 6edbdbfaefb215bbd75a90bfa1069f864104a9f7 Mon Sep 17 00:00:00 2001
From: limingshu <61349199+JamesLim-sy@users.noreply.github.com>
Date: Thu, 28 Oct 2021 09:19:40 +0800
Subject: [PATCH] first commit (#36778)

---
 python/paddle/fluid/optimizer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/paddle/fluid/optimizer.py b/python/paddle/fluid/optimizer.py
index 228ba084998..7412d3a3fe6 100755
--- a/python/paddle/fluid/optimizer.py
+++ b/python/paddle/fluid/optimizer.py
@@ -2068,6 +2068,7 @@ class LarsMomentumOptimizer(Optimizer):
             "lars_coeff": self._lars_coeff,
             "lars_weight_decay": [_lars_weight_decay],
             "multi_precision": find_master,
+            "epsilon": self._epsilon,
             "rescale_grad": self._rescale_grad
         }
 
-- 
GitLab