diff --git a/paddle/fluid/operators/checkpoint_notify_op.cc b/paddle/fluid/operators/checkpoint_notify_op.cc index c229cbf4984d91ef297e5cdd81977791fe8e5830..026ad722c275f21cc5689455e98a3cad7da8bfba 100644 --- a/paddle/fluid/operators/checkpoint_notify_op.cc +++ b/paddle/fluid/operators/checkpoint_notify_op.cc @@ -76,7 +76,7 @@ class CheckpointNotifyOpShapeInference : public framework::InferShapeBase { namespace ops = paddle::operators; -REGISTER_OPERATOR(checkpointnotify, ops::CheckpointNotifyOp, +REGISTER_OPERATOR(checkpoint_notify, ops::CheckpointNotifyOp, paddle::framework::EmptyGradOpMaker, ops::CheckpointNotifyOpMaker, ops::CheckpointNotifyOpShapeInference); diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py index bbd35aaecba27ea9fd66b9be585a972690980ab8..edc7ba69dd6843d6e16171c731b90c5965b71d41 100644 --- a/python/paddle/fluid/framework.py +++ b/python/paddle/fluid/framework.py @@ -382,7 +382,7 @@ class Operator(object): 'rnn_memory_helper_grad', 'conditional_block', 'while', 'send', 'recv', 'listen_and_serv', 'parallel_do', 'save_combine', 'load_combine', 'ncclInit', 'channel_create', 'channel_close', 'channel_send', - 'channel_recv', 'select' + 'channel_recv', 'select', 'checkpoint_notify' } def __init__(self, diff --git a/python/paddle/fluid/io.py b/python/paddle/fluid/io.py index 6a0e422cb3761312105df4a4d30b271eece2ff87..253fd5651c67df99ffbe7a0a8de8c52fe3f9270f 100644 --- a/python/paddle/fluid/io.py +++ b/python/paddle/fluid/io.py @@ -613,7 +613,7 @@ def save_pserver_vars_by_notify(executor, dirname, epmap): attrs['dir'] = cur_dir checkpoint_notify_block.append_op( - type='checkpointnotify', inputs={}, output={}, attrs=attrs) + type='checkpoint_notify', inputs={}, output={}, attrs=attrs) executor.run(checkpoint_notify_program)