load persistables with selected rows, test=develop (#17047)

6 years ago · 13295d90d9
parent 0c335dcd2c
commit 13295d90d9
1 changed files with 6 additions and 1 deletions
--- a/paddle/fluid/inference/io.cc
+++ b/paddle/fluid/inference/io.cc
@ -85,7 +85,12 @@ void LoadPersistables(framework::Executor* executor, framework::Scope* scope,
      new_var->SetShape(var->GetShape());
      new_var->SetDataType(var->GetDataType());
      new_var->SetType(var->GetType());
-      new_var->SetLoDLevel(var->GetLoDLevel());
+
+      if (var->GetType() !=
+          framework::proto::VarType::Type::VarType_Type_SELECTED_ROWS) {
+        new_var->SetLoDLevel(var->GetLoDLevel());
+      }
+
      new_var->SetPersistable(true);

      if (!param_filename.empty()) {