From b939904ca3c7ad28fce3b2926f1c45ba808cdca8 Mon Sep 17 00:00:00 2001 From: wukesong Date: Mon, 18 May 2020 21:48:07 +0800 Subject: [PATCH] remove enable_memreuse --- example/lenet_mnist/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/lenet_mnist/train.py b/example/lenet_mnist/train.py index 6c5cfb6fa2..3186f5fca7 100644 --- a/example/lenet_mnist/train.py +++ b/example/lenet_mnist/train.py @@ -40,7 +40,7 @@ if __name__ == "__main__": args = parser.parse_args() - context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target, enable_mem_reuse=False) + context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target) ds_train = create_dataset(os.path.join(args.data_path, "train"), cfg.batch_size, cfg.epoch_size)