From 1fb246a284acd0a60efad08340bc4807083a2f4e Mon Sep 17 00:00:00 2001
From: CaoJian <caojian05>
Date: Thu, 3 Dec 2020 19:53:39 +0800
Subject: [PATCH] set graph model should befor init(), otherwise hccl will
 report repeat init error.

---
 model_zoo/official/cv/vgg16/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model_zoo/official/cv/vgg16/train.py b/model_zoo/official/cv/vgg16/train.py
index 29bc344e574..57cc9da59cc 100644
--- a/model_zoo/official/cv/vgg16/train.py
+++ b/model_zoo/official/cv/vgg16/train.py
@@ -126,6 +126,7 @@ def merge_args(args_opt, cloud_args):
 if __name__ == '__main__':
     args = parse_args()
 
+    context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target)
     device_num = int(os.environ.get("DEVICE_NUM", 1))
     if args.is_distributed:
         if args.device_target == "Ascend":
@@ -143,7 +144,6 @@ if __name__ == '__main__':
     else:
         if args.device_target == "Ascend":
             context.set_context(device_id=args.device_id)
-    context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target)
 
     # select for master rank save ckpt or all rank save, compatible for model parallel
     args.rank_save_ckpt_flag = 0