提交 744e95d3 编写于 作者: T tangwei12

add ckpt load

上级 b677d821
...@@ -490,7 +490,7 @@ class Operator(object): ...@@ -490,7 +490,7 @@ class Operator(object):
'recv', 'listen_and_serv', 'parallel_do', 'save_combine', 'recv', 'listen_and_serv', 'parallel_do', 'save_combine',
'load_combine', 'ncclInit', 'channel_create', 'channel_close', 'load_combine', 'ncclInit', 'channel_create', 'channel_close',
'channel_send', 'channel_recv', 'select', 'checkpoint_save', 'channel_send', 'channel_recv', 'select', 'checkpoint_save',
'checkpoint_save' 'checkpoint_load'
} }
if type not in no_kernel_op_set: if type not in no_kernel_op_set:
self.desc.infer_var_type(self.block.desc) self.desc.infer_var_type(self.block.desc)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册