!6 remove two context param

Merge pull request !6 from jinyaohui/master

chapter03/lenet/main.py

@@ -88,7 +88,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)
 
     network = LeNet5(cfg.num_classes)
     net_loss = nn.SoftmaxCrossEntropyWithLogits(is_grad=False, sparse=True, reduction="mean")

chapter04/alexnet/main.py

@@ -78,7 +78,7 @@ if __name__ == "__main__":
     parser.add_argument('--dataset_sink_mode', type=bool, default=False, help='dataset_sink_mode is False or True')
     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)
 
     network = AlexNet(cfg.num_classes)
     loss = nn.SoftmaxCrossEntropyWithLogits(is_grad=False, sparse=True, reduction="mean")

chapter05/resnet/resnet_cifar.py

@@ -66,8 +66,6 @@ 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(device_id=device_id)
-    context.set_context(enable_loop_sink=True)
-    context.set_context(enable_mem_reuse=False)
 
 def create_dataset(repeat_num=1, training=True):
     """create the dataset of cifar10"""

chapter07/Bert_NEZHA_cnwiki/train.py

@@ -74,8 +74,6 @@ def train_bert():
     """train bert"""
     context.set_context(mode=context.GRAPH_MODE)
     context.set_context(device_target="Ascend")
-    context.set_context(enable_loop_sink=True)
-    context.set_context(enable_mem_reuse=True)
     ds = create_train_dataset(bert_net_cfg.batch_size)
     netwithloss = BertNetworkWithLoss(bert_net_cfg, True)
     optimizer = Lamb(netwithloss.trainable_params(), decay_steps=bert_train_cfg.decay_steps,