We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f892f4d commit 1ba79a5Copy full SHA for 1ba79a5
dygraph/cycle_gan/test.py
@@ -38,7 +38,7 @@ def test():
38
os.makedirs(out_path)
39
cycle_gan = Cycle_Gan("cycle_gan")
40
save_dir = args.init_model + str(epoch)
41
- restore = fluid.dygraph.load_persistables(save_dir)
+ restore, _ = fluid.dygraph.load_persistables(save_dir)
42
cycle_gan.load_dict(restore)
43
cycle_gan.eval()
44
for data_A , data_B in zip(A_test_reader(), B_test_reader()):
@@ -74,4 +74,4 @@ def test():
74
if __name__ == "__main__":
75
args = parser.parse_args()
76
print_arguments(args)
77
- test()
+ test()
0 commit comments