fix load bug (PaddlePaddle#2435)

load_persistables function returns a tuple which include parameters and optimizer
nuaaceieyty · Jun 18, 2019 · dc8813a · dc8813a
1 parent 7935b52
commit dc8813a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dygraph/cycle_gan/infer.py b/dygraph/cycle_gan/infer.py
@@ -38,7 +38,7 @@ def infer():
             os.makedirs(out_path)
         cycle_gan = Cycle_Gan("cycle_gan")
         save_dir = args.init_model 
-        restore = fluid.dygraph.load_persistables(save_dir)
+        restore, _ = fluid.dygraph.load_persistables(save_dir)
         cycle_gan.load_dict(restore)
         cycle_gan.eval()
         for file in glob.glob(args.input):