remove enable_memreuse

This commit is contained in:
wukesong 2020-05-18 21:48:07 +08:00
parent c77e82b811
commit b939904ca3
1 changed files with 1 additions and 1 deletions

View File

@ -40,7 +40,7 @@ if __name__ == "__main__":
args = parser.parse_args() 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"), ds_train = create_dataset(os.path.join(args.data_path, "train"),
cfg.batch_size, cfg.batch_size,
cfg.epoch_size) cfg.epoch_size)