From c80125f286fb641472b62a51c6f350e00e904519 Mon Sep 17 00:00:00 2001
From: tangwei12 <tangwei12@baidu.com>
Date: Tue, 15 May 2018 17:16:17 +0800
Subject: [PATCH] add checkpoint_load to python framework

---
 python/paddle/fluid/framework.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py
index 461226354..6ab31ec94 100644
--- 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)
-- 
GitLab