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

revert-31562-mean
wangguanzhong 4 years ago committed by GitHub
parent c3c064a8fc
commit 844d8e0c2c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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",

Loading…
Cancel
Save