fix demo bug in unstructured pruner (#895)

b3d82cb5 · minghaoBD · GitHub · c87b7f4c · b3d82cb5
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

demo/unstructured_prune/train.py demo/unstructured_prune/train.py +1 -1

未找到文件。
--- a/demo/unstructured_prune/train.py
+++ b/demo/unstructured_prune/train.py
@@ -325,7 +325,7 @@ def compress(args):
                fleet.save_persistables(executor=exe, dirname=args.model_path)
            else:
                paddle.fluid.io.save_persistables(
-                    executor, dirname=args.model_path)
+                    executor=exe, dirname=args.model_path)
 def main():