提交 3dd27465 编写于 作者: T tangwei12

modify Get -> GetMutable

上级 955c7934
...@@ -86,11 +86,11 @@ class CheckpointLoadOp : public framework::OperatorBase { ...@@ -86,11 +86,11 @@ class CheckpointLoadOp : public framework::OperatorBase {
var_file.append(inp_var_names[i]); var_file.append(inp_var_names[i]);
VLOG(3) << "ready to load var: " << inp_var_names[i]; VLOG(3) << "ready to load var: " << inp_var_names[i];
auto &tensor = var->Get<framework::LoDTensor>(); auto *tensor = var->GetMutable<framework::LoDTensor>();
std::ifstream fin(var_file); std::ifstream fin(var_file);
PADDLE_ENFORCE(static_cast<bool>(fin), "Cannot open file %s for load op", PADDLE_ENFORCE(static_cast<bool>(fin), "Cannot open file %s for load op",
var_file); var_file);
DeserializeFromStream(fin, tensor, dev_ctx); framework::DeserializeFromStream(fin, tensor, dev_ctx);
fin.close(); fin.close();
VLOG(3) << " load var: " << inp_var_names[i] << " finished"; VLOG(3) << " load var: " << inp_var_names[i] << " finished";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册