未验证 提交 4348fc35 编写于 作者: D Double_V 提交者: GitHub

Merge pull request #5803 from WenmuZhou/pse

rm view
...@@ -121,9 +121,9 @@ class PSELoss(nn.Layer): ...@@ -121,9 +121,9 @@ class PSELoss(nn.Layer):
if neg_num == 0: if neg_num == 0:
selected_mask = training_mask selected_mask = training_mask
selected_mask = selected_mask.view( selected_mask = selected_mask.reshape(
1, selected_mask.shape[0], [1, selected_mask.shape[0], selected_mask.shape[1]]).astype(
selected_mask.shape[1]).astype('float32') 'float32')
return selected_mask return selected_mask
neg_score = paddle.masked_select(score, gt_text <= 0.5) neg_score = paddle.masked_select(score, gt_text <= 0.5)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册