diff --git a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_full_1p.sh b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_full_1p.sh
index 81418499ea3d9778c1456d196a45a530c44ccd24..37d00d1e17286d79349c204c7d10c8a74f2857f0 100644
--- a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_full_1p.sh
+++ b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_full_1p.sh
@@ -114,13 +114,13 @@ batch_size=16
 if [ x"${modelarts_flag}" != x ];
 then
     python3.7 ./train_ultrafast.py \
-        --data-path=${data_path}/data/llamas \
+        --data_path=${data_path}/data/llamas \
         --output_path=${output_path} \
         --epochs=${train_epochs} \
         --batch_size=${batch_size} 1>${print_log} 2>&1
 else
     python3.7 ./train_ultrafast.py \
-        --data-path=${data_path}/data/llamas \
+        --data_path=${data_path}/data/llamas \
         --output_path=${output_path} \
         --epochs=${train_epochs} \
         --batch_size=${batch_size} 1>${print_log} 2>&1
diff --git a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_performance_1p.sh b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_performance_1p.sh
index 81418499ea3d9778c1456d196a45a530c44ccd24..37d00d1e17286d79349c204c7d10c8a74f2857f0 100644
--- a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_performance_1p.sh
+++ b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_performance_1p.sh
@@ -114,13 +114,13 @@ batch_size=16
 if [ x"${modelarts_flag}" != x ];
 then
     python3.7 ./train_ultrafast.py \
-        --data-path=${data_path}/data/llamas \
+        --data_path=${data_path}/data/llamas \
         --output_path=${output_path} \
         --epochs=${train_epochs} \
         --batch_size=${batch_size} 1>${print_log} 2>&1
 else
     python3.7 ./train_ultrafast.py \
-        --data-path=${data_path}/data/llamas \
+        --data_path=${data_path}/data/llamas \
         --output_path=${output_path} \
         --epochs=${train_epochs} \
         --batch_size=${batch_size} 1>${print_log} 2>&1
diff --git a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py
index d0d38ad1643cb1fc20a42b91d040ed010d515854..5c44a134245e0bc91e2049cecda7d9ba2407b4da 100644
--- a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py
+++ b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py
@@ -28,8 +28,8 @@
 
 
 import npu_device as npu
-npu_device.global_options().op_compiler_cache_mode="enable"
-npu_device.global_options().op_compiler_cache_dir="/mnt/home/test_user08/UltraFast_NPU/my_kernel_cache"
+npu.global_options().op_compiler_cache_mode="enable"
+npu.global_options().op_compiler_cache_dir="/mnt/home/test_user08/UltraFast_NPU/my_kernel_cache"
 npu.open().as_default()
 
 import tensorflow as tf
@@ -45,15 +45,15 @@ from utils.datasets import llamas_dataset, labelme_dataset
 
 parser = argparse.ArgumentParser(description='Train Ultrafast Net')
 parser.add_argument('--epochs', type=int, default=1)
-parser.add_argument('--batch-size', type=int, default=16)
-parser.add_argument('--max-images', type=int, default=None)
-parser.add_argument('--prefetch-size', type=int, default=256)
-parser.add_argument('--resnet-weights', default=None)
-parser.add_argument('--base-path', default='')
-parser.add_argument('--llamas-path', '--data-path', default='./data/llamas/')
-parser.add_argument('--labelme-path', default=[], action='append')
-parser.add_argument('--output-path', default='./')
-parser.add_argument('--model-name', default='ultrafast')
+parser.add_argument('--batch_size', type=int, default=16)
+parser.add_argument('--max_images', type=int, default=None)
+parser.add_argument('--prefetch_size', type=int, default=256)
+parser.add_argument('--resnet_weights', default=None)
+parser.add_argument('--base_path', default='')
+parser.add_argument('--llamas_path', '--data_path', default='./data/llamas/')
+parser.add_argument('--labelme_path', default=[], action='append')
+parser.add_argument('--output_path', default='./')
+parser.add_argument('--model_name', default='ultrafast')
 
 args = parser.parse_args()
 print(args)
diff --git a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/utils/datasets.py b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/utils/datasets.py
index da95e1c5077ce4b33c66d29ab0d597d052419227..d0266f3989d100ebd67da28588bbbb657360e02d 100644
--- a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/utils/datasets.py
+++ b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/utils/datasets.py
@@ -189,7 +189,7 @@ def load_json_dataset(json_file_pattern, processor, max_records=None, shuffle_si
     if premap_func is not None:
         ds = premap_func(ds)
         
-    ds = ds.map(processor.process_json, num_parallel_calls=64)
+    ds = ds.map(processor.process_json, num_parallel_calls=32)
     return ds