diff --git a/paddle/fluid/operators/checkpoint_load_op.cc b/paddle/fluid/operators/checkpoint_load_op.cc index 8edf3b6429dbddfaa1c78d7953595f655bee3b85..ec451c9f3f0cd28a76442e136c3c91a8fa9f39c5 100644 --- a/paddle/fluid/operators/checkpoint_load_op.cc +++ b/paddle/fluid/operators/checkpoint_load_op.cc @@ -61,6 +61,13 @@ class CheckpointLoadOpProtoMaker : public framework::OpProtoAndCheckerMaker { public: CheckpointLoadOpProtoMaker(OpProto *proto, OpAttrChecker *op_checker) : OpProtoAndCheckerMaker(proto, op_checker) { + AddComment(R"DOC( +CheckpointLoad operator + +This operator will serialize and write a list of input LoDTensor variables +to a file on disk. +)DOC"); + AddAttr( "dir", "(string)" diff --git a/paddle/fluid/operators/checkpoint_save_op.cc b/paddle/fluid/operators/checkpoint_save_op.cc index 7449352117b58a4cb7d89edacb5211b4c5edb9ed..1082bb4a345a2ede31642c228922be80dd075c36 100644 --- a/paddle/fluid/operators/checkpoint_save_op.cc +++ b/paddle/fluid/operators/checkpoint_save_op.cc @@ -180,7 +180,7 @@ class CheckpointSaveOpShapeInference : public framework::InferShapeBase { namespace ops = paddle::operators; -REGISTER_OPERATOR(send_vars, ops::CheckpointSaveOp, +REGISTER_OPERATOR(checkpoint_save, ops::CheckpointSaveOp, paddle::framework::EmptyGradOpMaker, ops::CheckpointSaveOpProtoMaker, ops::CheckpointSaveOpVarTypeInference,