diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py index 185f4f6651f816e5588ebed31123ae5f84f9ca3b..fbe4531f056a583d2b4d855d32fe8e04da94fa3c 100644 --- a/python/paddle/fluid/framework.py +++ b/python/paddle/fluid/framework.py @@ -965,6 +965,13 @@ class Block(object): if var.type == core.VarDesc.VarType.STEP_SCOPES: ret_var = self.create_var( name=var.name, persistable=var.persistable, type=var.type) + elif var.type == core.VarDesc.VarType.SELECTED_ROWS: + ret_var = self.create_var( + name=var.name, + shape=var.shape, + dtype=var.dtype, + type=var.type, + persistable=True) else: ret_var = self.create_var( name=var.name,