未验证 提交 844d8e0c 编写于 作者: W wangguanzhong 提交者: GitHub

add REGISTER_OP_VERSION for generate_proposals, roi_align, roi_pool test=op_version (#30034)

上级 c3c064a8
......@@ -303,6 +303,13 @@ REGISTER_OPERATOR(
REGISTER_OP_CPU_KERNEL(generate_proposals, ops::GenerateProposalsKernel<float>,
ops::GenerateProposalsKernel<double>);
REGISTER_OP_VERSION(generate_proposals)
.AddCheckpoint(
R"ROC(
Incompatible upgrade of output [RpnRoisLod])ROC",
paddle::framework::compatible::OpVersionDesc().DeleteOutput(
"RpnRoisLod",
"Delete RpnRoisLod due to incorrect output name and "
"it is not used in object detection models yet."))
.AddCheckpoint(
R"ROC(
Upgrade generate_proposals add a new output [RpnRoisNum])ROC",
......
......@@ -233,6 +233,13 @@ REGISTER_OP_CPU_KERNEL(
ops::CPUROIAlignGradOpKernel<paddle::platform::CPUDeviceContext, double>,
ops::CPUROIAlignGradOpKernel<paddle::platform::CPUDeviceContext, int>);
REGISTER_OP_VERSION(roi_align)
.AddCheckpoint(
R"ROC(
Incompatible upgrade of input [RpnRoisLod])ROC",
paddle::framework::compatible::OpVersionDesc().DeleteInput(
"RpnRoisLod",
"Delete RpnRoisLod due to incorrect input name and "
"it is not used in object detection models yet."))
.AddCheckpoint(
R"ROC(
Upgrade roi_align add a new input [RoisNum])ROC",
......
......@@ -227,6 +227,13 @@ REGISTER_OP_CPU_KERNEL(
ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, double>,
ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, int>);
REGISTER_OP_VERSION(roi_pool)
.AddCheckpoint(
R"ROC(
Incompatible upgrade of input [RpnRoisLod])ROC",
paddle::framework::compatible::OpVersionDesc().DeleteInput(
"RpnRoisLod",
"Delete RpnRoisLod due to incorrect input name and "
"it is not used in object detection models yet."))
.AddCheckpoint(
R"ROC(
Upgrade roi_pool add a new input [RoisNum])ROC",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册