diff --git a/official/cv/simple_pose/default_config.yaml b/official/cv/simple_pose/default_config.yaml
index de5fbbeed0144230777a56780fbef076ad5d3ede..09fe2e1741305b8630b36fb3160dec7bbf83fcfb 100644
--- a/official/cv/simple_pose/default_config.yaml
+++ b/official/cv/simple_pose/default_config.yaml
@@ -54,8 +54,8 @@ DATASET:
SCALE_FACTOR: 0.3
#train-related
TRAIN:
- SINK_MODE: False
- NUM_WORKERS: 8
+ SINK_MODE: True
+ NUM_WORKERS: 16
BATCH_SIZE: 64
BEGIN_EPOCH: 0
END_EPOCH: 140
diff --git a/official/cv/simple_pose/train.py b/official/cv/simple_pose/train.py
index 165f029fa02f65457ec9ca1248596f3a778b315a..c8e7db7ce07f1a4c68d05302d8e3d225d0ee0b52 100644
--- a/official/cv/simple_pose/train.py
+++ b/official/cv/simple_pose/train.py
@@ -125,7 +125,7 @@ def modelarts_pre_process():
def run_train():
print('batch size :{}'.format(config.TRAIN.BATCH_SIZE))
if config.run_distribute:
- init("nccl")
+ init()
rank = get_rank()
device_num = get_device_num()
context.set_auto_parallel_context(device_num=device_num,
diff --git a/research/cv/SPPNet/train.py b/research/cv/SPPNet/train.py
index a74aaebf9d99411ce2fb0c5d2d02a1747db09a52..9cb3497eb42619ea89887f444bea3a6b067742db 100644
--- a/research/cv/SPPNet/train.py
+++ b/research/cv/SPPNet/train.py
@@ -78,7 +78,7 @@ if __name__ == "__main__":
if args.is_distributed:
if args.device_target == "Ascend":
- context.set_context(device_id=device_id)
+ context.set_context(device_id=args.device_id)
init("hccl")
else:
assert args.device_target == "GPU"