From b9c03eac17ed7b4e0f0442eaa1fccbd464a975e2 Mon Sep 17 00:00:00 2001
From: sandyhouse <lilong12@baidu.com>
Date: Sun, 7 Mar 2021 20:43:57 +0800
Subject: [PATCH] update

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

diff --git a/python/paddle/fluid/optimizer.py b/python/paddle/fluid/optimizer.py
index 633c8f3ab71..4d5773c7675 100644
--- a/python/paddle/fluid/optimizer.py
+++ b/python/paddle/fluid/optimizer.py
@@ -4723,7 +4723,7 @@ class PipelineOptimizer(object):
                     extra_index += 1
 
                     fill_shape = list(var.shape)
-                    fill_shape[0] = 4
+                    fill_shape[0] = self.pp_bz
                     block._insert_op_without_sync(
                         index=index + extra_index,
                         type='fill_constant',
-- 
GitLab