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

fix generate_mask_labels lod level (#22743)

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