未验证 提交 ac7cd54c 编写于 作者: L littletomatodonkey 提交者: GitHub

remove persistable (#941)

上级 8055e86e
...@@ -96,7 +96,6 @@ class BlazeFace(object): ...@@ -96,7 +96,6 @@ class BlazeFace(object):
overlap_threshold=0.35, overlap_threshold=0.35,
neg_overlap=0.35) neg_overlap=0.35)
loss = fluid.layers.reduce_sum(loss) loss = fluid.layers.reduce_sum(loss)
loss.persistable = True
return {'loss': loss} return {'loss': loss}
else: else:
pred = self.output_decoder(locs, confs, box, box_var) pred = self.output_decoder(locs, confs, box, box_var)
......
...@@ -87,7 +87,6 @@ class FaceBoxes(object): ...@@ -87,7 +87,6 @@ class FaceBoxes(object):
overlap_threshold=0.35, overlap_threshold=0.35,
neg_overlap=0.35) neg_overlap=0.35)
loss = fluid.layers.reduce_sum(loss) loss = fluid.layers.reduce_sum(loss)
loss.persistable = True
return {'loss': loss} return {'loss': loss}
else: else:
pred = self.output_decoder(locs, confs, box, box_var) pred = self.output_decoder(locs, confs, box, box_var)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册