diff --git a/PyTorch/built-in/cv/classification/ResNet50_ID4149_for_PyTorch/test/train_performance_8p.sh b/PyTorch/built-in/cv/classification/ResNet50_ID4149_for_PyTorch/test/train_performance_8p.sh
index 46bf1b47d19e4f93b01b8ae49c5bb8b64f189ac4..29cd43a97b675dd153df3221fbec252e31a9a57a 100644
--- a/PyTorch/built-in/cv/classification/ResNet50_ID4149_for_PyTorch/test/train_performance_8p.sh
+++ b/PyTorch/built-in/cv/classification/ResNet50_ID4149_for_PyTorch/test/train_performance_8p.sh
@@ -29,6 +29,9 @@ do
     elif [[ $para == --fp32 ]];then
       	fp32=`echo ${para#*=}`
       	export ALLOW_FP32=True
+    elif [[ $para == --profiling* ]];then
+        profiling=`echo ${para#*=}`
+        export PROFILE_TYPE=${profiling}
     fi
 done