diff --git a/PaddleCV/deeplabv3+/train.py b/PaddleCV/deeplabv3+/train.py index 5e983ed291f2b434a79ecefa7a3583f2741bd0ab..2cef945de72311f78ca313c37dd25f820fd7d3e2 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,