enable lookup table to inference

7 years ago · 057a64508c
parent f1a10ccea9
commit 057a64508c
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/fluid/io.py
+++ b/python/paddle/fluid/io.py
@ -370,6 +370,7 @@ def load_vars(executor,
        load_vars(
            executor,
            dirname=dirname,
+            main_program=main_program,
            vars=list(filter(predicate, main_program.list_vars())),
            filename=filename)
    else: