From ccc5418841f263dba9d89add8ac0cd3baf55bc4a Mon Sep 17 00:00:00 2001 From: Yancey Date: Fri, 9 Mar 2018 19:28:17 +0800 Subject: [PATCH] Don't save var which type is RAW (#8928) --- python/paddle/fluid/io.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/paddle/fluid/io.py b/python/paddle/fluid/io.py index 1817caa9427..35aa80a2ae9 100644 --- a/python/paddle/fluid/io.py +++ b/python/paddle/fluid/io.py @@ -102,6 +102,9 @@ def save_vars(executor, save_var_map = {} for each_var in vars: + # NOTE: don't save the variable which type is RAW + if each_var.type == core.VarDesc.VarType.RAW: + continue new_var = _clone_var_in_block_(save_block, each_var) if filename is None: save_block.append_op( -- GitLab