diff --git a/ppdet/modeling/architectures/blazeface.py b/ppdet/modeling/architectures/blazeface.py index 245a43835fcda15ef541ae7c42f2163042ade411..62230cc5f316b0a33fcfbcd720a02eed1f907eaa 100644 --- a/ppdet/modeling/architectures/blazeface.py +++ b/ppdet/modeling/architectures/blazeface.py @@ -96,7 +96,6 @@ class BlazeFace(object): overlap_threshold=0.35, neg_overlap=0.35) loss = fluid.layers.reduce_sum(loss) - loss.persistable = True return {'loss': loss} else: pred = self.output_decoder(locs, confs, box, box_var) diff --git a/ppdet/modeling/architectures/faceboxes.py b/ppdet/modeling/architectures/faceboxes.py index f5b120ecef430973d95053742cea282fbb729f34..b55354902238bb960e6ebf9605ec42c130e27b38 100644 --- a/ppdet/modeling/architectures/faceboxes.py +++ b/ppdet/modeling/architectures/faceboxes.py @@ -87,7 +87,6 @@ class FaceBoxes(object): overlap_threshold=0.35, neg_overlap=0.35) loss = fluid.layers.reduce_sum(loss) - loss.persistable = True return {'loss': loss} else: pred = self.output_decoder(locs, confs, box, box_var)