!1233 remove params "enable_mem_reuse"

Merge pull request !1233 from wukesong/remove-memreuse
5 years ago · d8c5daf981
parent 2cc9fd1cf4 b939904ca3
commit d8c5daf981
1 changed files with 1 additions and 1 deletions
--- 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)