提交 459690ae 编写于 作者: T tangwei12

bug fux

上级 620698e7
...@@ -132,11 +132,8 @@ class SaveOp : public framework::OperatorBase { ...@@ -132,11 +132,8 @@ class SaveOp : public framework::OperatorBase {
void SaveSelectedRows(const framework::Scope &scope, void SaveSelectedRows(const framework::Scope &scope,
const platform::Place &place, const platform::Place &place,
framework::Variable *var) const { framework::Variable *var) const {
auto *lt_var = scope.FindVar("loopup_table_path")->GetMutable<std::string>();
auto lt_varname = string::Sprintf("%s.path", Input("X")); PADDLE_ENFORCE(lt_var != nullptr, "Cannot find variable loopup_table_path for SaveSelectedRows");
auto *lt_var = scope.FindVar(lt_varname)->GetMutable<std::string>();
PADDLE_ENFORCE(lt_var != nullptr, "Cannot find variable %s for SaveSelectedRows",
lt_varname);
std::string filename = lt_var->data(); std::string filename = lt_var->data();
VLOG(4) << "SaveSelectedRows get File name: " << filename; VLOG(4) << "SaveSelectedRows get File name: " << filename;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册