Update test.py (#2408)

fix load bug

Update test.py (#2408)
fix load bug
1ba79a58 · Divano · lvmengsi · f892f4d9 · 1ba79a58
显示空白变更内容
内联并排

Showing with 2 addition and 2 deletion

dygraph/cycle_gan/test.py dygraph/cycle_gan/test.py +2 -2

未找到文件。
--- a/dygraph/cycle_gan/test.py
+++ b/dygraph/cycle_gan/test.py
@@ -38,7 +38,7 @@ def test():
            os.makedirs(out_path)
        cycle_gan = Cycle_Gan("cycle_gan")
        save_dir = args.init_model + str(epoch)
-        restore = fluid.dygraph.load_persistables(save_dir)
+        restore, _ = fluid.dygraph.load_persistables(save_dir)
        cycle_gan.load_dict(restore)
        cycle_gan.eval()
        for data_A , data_B in zip(A_test_reader(), B_test_reader()):