fix some bug in prune demo (#572)

demo/prune/train.py

@@ -154,7 +154,7 @@ def compress(args):
         batch_size=args.batch_size,
         shuffle=True,
         return_list=False,
-        use_shared_memory=False,
+        use_shared_memory=True,
         num_workers=16)
     valid_loader = paddle.io.DataLoader(
         val_dataset,
@@ -162,7 +162,7 @@ def compress(args):
         feed_list=[image, label],
         drop_last=False,
         return_list=False,
-        use_shared_memory=False,
+        use_shared_memory=True,
         batch_size=args.batch_size,
         shuffle=False)
 
@@ -245,7 +245,7 @@ def compress(args):
             infer_model_path = os.path.join(args.model_path, "infer_models",
                                             str(i))
             paddle.static.save_inference_model(
-                infer_model_path, ["image"], [out],
+                infer_model_path, [image], [out],
                 exe,
                 program=pruned_val_program)
             _logger.info("Saved inference model into [{}]".format(

demo/sensitive/train.py

@@ -75,6 +75,7 @@ def compress(args):
         feed_list=[image, label],
         drop_last=False,
         batch_size=args.batch_size,
+        use_shared_memory=True,
         shuffle=False)
 
     def test(program):