提交 d8c5daf9 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!1233 remove params "enable_mem_reuse"

Merge pull request !1233 from wukesong/remove-memreuse
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册