提交 0334d494 编写于 作者: T tangwei12

fix bug

上级 30b50dcf
...@@ -61,6 +61,13 @@ class CheckpointLoadOpProtoMaker : public framework::OpProtoAndCheckerMaker { ...@@ -61,6 +61,13 @@ class CheckpointLoadOpProtoMaker : public framework::OpProtoAndCheckerMaker {
public: public:
CheckpointLoadOpProtoMaker(OpProto *proto, OpAttrChecker *op_checker) CheckpointLoadOpProtoMaker(OpProto *proto, OpAttrChecker *op_checker)
: OpProtoAndCheckerMaker(proto, 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<std::string>( AddAttr<std::string>(
"dir", "dir",
"(string)" "(string)"
......
...@@ -180,7 +180,7 @@ class CheckpointSaveOpShapeInference : public framework::InferShapeBase { ...@@ -180,7 +180,7 @@ class CheckpointSaveOpShapeInference : public framework::InferShapeBase {
namespace ops = paddle::operators; namespace ops = paddle::operators;
REGISTER_OPERATOR(send_vars, ops::CheckpointSaveOp, REGISTER_OPERATOR(checkpoint_save, ops::CheckpointSaveOp,
paddle::framework::EmptyGradOpMaker, paddle::framework::EmptyGradOpMaker,
ops::CheckpointSaveOpProtoMaker, ops::CheckpointSaveOpProtoMaker,
ops::CheckpointSaveOpVarTypeInference, ops::CheckpointSaveOpVarTypeInference,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册