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

!5 remove enable_task_sink param

Merge pull request !5 from gengdongjie/master
......@@ -65,7 +65,7 @@ context.set_context(mode=context.GRAPH_MODE, device_target=args_opt.device_targe
if args_opt.device_target == "Ascend":
#Choose one availabe Device to use on users' env.
device_id = int(os.getenv('DEVICE_ID'))
context.set_context(enable_task_sink=True, device_id=device_id)
context.set_context(device_id=device_id)
context.set_context(enable_loop_sink=True)
context.set_context(enable_mem_reuse=False)
......
......@@ -74,7 +74,6 @@ def train_bert():
"""train bert"""
context.set_context(mode=context.GRAPH_MODE)
context.set_context(device_target="Ascend")
context.set_context(enable_task_sink=True)
context.set_context(enable_loop_sink=True)
context.set_context(enable_mem_reuse=True)
ds = create_train_dataset(bert_net_cfg.batch_size)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册