bug fix

7 years ago · cd98f2b7e0
parent eff92d046b
commit cd98f2b7e0
2 changed files with 2 additions and 4 deletions
--- a/paddle/fluid/operators/checkpoint_load_op.cc
+++ b/paddle/fluid/operators/checkpoint_load_op.cc
@ -154,8 +154,7 @@ class CheckpointLoadOp : public framework::OperatorBase {

 class CheckpointLoadOpProtoMaker : public framework::OpProtoAndCheckerMaker {
 public:
-  CheckpointLoadOpProtoMaker(OpProto *proto, OpAttrChecker *op_checker)
-      : OpProtoAndCheckerMaker(proto, op_checker) {
+  void Make() override {
    AddInput(
        "X",
        "(vector) Input LoDTensors that need to be saved together in a file.")
--- a/paddle/fluid/operators/checkpoint_save_op.cc
+++ b/paddle/fluid/operators/checkpoint_save_op.cc
@ -143,8 +143,7 @@ class CheckpointSaveOp : public framework::OperatorBase {

 class CheckpointSaveOpProtoMaker : public framework::OpProtoAndCheckerMaker {
 public:
-  CheckpointSaveOpProtoMaker(OpProto *proto, OpAttrChecker *op_checker)
-      : OpProtoAndCheckerMaker(proto, op_checker) {
+  void Make() override {
    AddInput(
        "X",
        "(vector) Input LoDTensors that need to be saved together in a file.")