From cd98f2b7e0eb251659565c9f9171f52c95c819f8 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Fri, 18 May 2018 17:44:48 +0800 Subject: [PATCH] bug fix --- paddle/fluid/operators/checkpoint_load_op.cc | 3 +-- paddle/fluid/operators/checkpoint_save_op.cc | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/paddle/fluid/operators/checkpoint_load_op.cc b/paddle/fluid/operators/checkpoint_load_op.cc index a9676de36..82a40e18d 100644 --- 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.") diff --git a/paddle/fluid/operators/checkpoint_save_op.cc b/paddle/fluid/operators/checkpoint_save_op.cc index 30eda30c5..790fd4ea6 100644 --- 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.") -- GitLab