未验证 提交 1c065346 编写于 作者: D dyning 提交者: GitHub

fix generate_mask_labels lod level (#22743)

上级 ba140222
...@@ -75,6 +75,11 @@ class GenerateMaskLabelsOp : public framework::OperatorWithKernel { ...@@ -75,6 +75,11 @@ class GenerateMaskLabelsOp : public framework::OperatorWithKernel {
ctx->SetOutputDim("MaskRois", {-1, 4}); ctx->SetOutputDim("MaskRois", {-1, 4});
ctx->SetOutputDim("RoiHasMaskInt32", {-1, 1}); ctx->SetOutputDim("RoiHasMaskInt32", {-1, 1});
ctx->SetOutputDim("MaskInt32", {-1, num_classes * resolution * resolution}); ctx->SetOutputDim("MaskInt32", {-1, num_classes * resolution * resolution});
if (!ctx->IsRuntime()) {
ctx->SetLoDLevel("MaskRois", ctx->GetLoDLevel("Rois"));
ctx->SetLoDLevel("RoiHasMaskInt32", ctx->GetLoDLevel("Rois"));
ctx->SetLoDLevel("MaskInt32", ctx->GetLoDLevel("Rois"));
}
} }
protected: protected:
......
...@@ -21,7 +21,6 @@ ...@@ -21,7 +21,6 @@
COMPILE_RUN_OP_WHITE_LIST = [ COMPILE_RUN_OP_WHITE_LIST = [
'sequence_pool', \ 'sequence_pool', \
'sequence_slice', \ 'sequence_slice', \
'generate_mask_labels', \
'generate_proposals', \ 'generate_proposals', \
'mine_hard_examples', \ 'mine_hard_examples', \
'retinanet_detection_output', \ 'retinanet_detection_output', \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册