From 7a372767c7aceddb9575fc793e93146d20a5b9b0 Mon Sep 17 00:00:00 2001
From: Bai Yifan <bai.yf@qq.com>
Date: Thu, 16 May 2019 17:39:55 +0800
Subject: [PATCH] fix warnings,test=develop (#2261)

---
 PaddleCV/deeplabv3+/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/PaddleCV/deeplabv3+/train.py b/PaddleCV/deeplabv3+/train.py
index 5e983ed2..2cef945d 100755
--- a/PaddleCV/deeplabv3+/train.py
+++ b/PaddleCV/deeplabv3+/train.py
@@ -145,6 +145,7 @@ with fluid.program_guard(tp, sp):
         fluid.layers.assign(np.array(
             [0.1], dtype=np.float32)))
     loss_mean = fluid.layers.reduce_mean(loss) / area
+    loss_mean.persistable = True
 
     opt = fluid.optimizer.Momentum(
         lr,
-- 
GitLab