add checkpoint_load to python framework

7 years ago · c80125f286
parent f04b23adf9
commit c80125f286
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/framework.py
+++ b/python/paddle/fluid/framework.py
@ -489,7 +489,8 @@ 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', 'checkpoint_save'
+            'channel_send', 'channel_recv', 'select', 'checkpoint_save',
+            'checkpoint_save'
        }
        if type not in no_kernel_op_set:
            self.desc.infer_var_type(self.block.desc)