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 7935b52 commit dc8813aCopy full SHA for dc8813a
dygraph/cycle_gan/infer.py
@@ -38,7 +38,7 @@ def infer():
38
os.makedirs(out_path)
39
cycle_gan = Cycle_Gan("cycle_gan")
40
save_dir = args.init_model
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 file in glob.glob(args.input):
0 commit comments