Browse Source

remove enable_memreuse

tags/v0.3.0-alpha
wukesong 5 years ago
parent
commit
b939904ca3
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      example/lenet_mnist/train.py

+ 1
- 1
example/lenet_mnist/train.py View File

@@ -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)


Loading…
Cancel
Save