Skip to content

Commit a3300bc

Browse files
committed
add REGISTER_OP_VERSION for generate_proposals, roi_align, roi_pool test=op_version
1 parent 65d4ff7 commit a3300bc

File tree

3 files changed

+21
-0
lines changed

3 files changed

+21
-0
lines changed

paddle/fluid/operators/detection/generate_proposals_op.cc

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -303,6 +303,13 @@ REGISTER_OPERATOR(
303303
REGISTER_OP_CPU_KERNEL(generate_proposals, ops::GenerateProposalsKernel<float>,
304304
ops::GenerateProposalsKernel<double>);
305305
REGISTER_OP_VERSION(generate_proposals)
306+
.AddCheckpoint(
307+
R"ROC(
308+
Incompatible upgrade of output [RpnRoisLod])ROC",
309+
paddle::framework::compatible::OpVersionDesc().DeleteOutput(
310+
"RpnRoisLod",
311+
"Delete RpnRoisLod due to incorrect output name and "
312+
"it is not used in object detection models yet."))
306313
.AddCheckpoint(
307314
R"ROC(
308315
Upgrade generate_proposals add a new output [RpnRoisNum])ROC",

paddle/fluid/operators/roi_align_op.cc

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -233,6 +233,13 @@ REGISTER_OP_CPU_KERNEL(
233233
ops::CPUROIAlignGradOpKernel<paddle::platform::CPUDeviceContext, double>,
234234
ops::CPUROIAlignGradOpKernel<paddle::platform::CPUDeviceContext, int>);
235235
REGISTER_OP_VERSION(roi_align)
236+
.AddCheckpoint(
237+
R"ROC(
238+
Incompatible upgrade of input [RpnRoisLod])ROC",
239+
paddle::framework::compatible::OpVersionDesc().DeleteInput(
240+
"RpnRoisLod",
241+
"Delete RpnRoisLod due to incorrect input name and "
242+
"it is not used in object detection models yet."))
236243
.AddCheckpoint(
237244
R"ROC(
238245
Upgrade roi_align add a new input [RoisNum])ROC",

paddle/fluid/operators/roi_pool_op.cc

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,13 @@ REGISTER_OP_CPU_KERNEL(
227227
ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, double>,
228228
ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, int>);
229229
REGISTER_OP_VERSION(roi_pool)
230+
.AddCheckpoint(
231+
R"ROC(
232+
Incompatible upgrade of input [RpnRoisLod])ROC",
233+
paddle::framework::compatible::OpVersionDesc().DeleteInput(
234+
"RpnRoisLod",
235+
"Delete RpnRoisLod due to incorrect input name and "
236+
"it is not used in object detection models yet."))
230237
.AddCheckpoint(
231238
R"ROC(
232239
Upgrade roi_pool add a new input [RoisNum])ROC",

0 commit comments

Comments
 (0)