From 0cdaaf295381cfa946b9c5f1a6e18a2862dc41bb Mon Sep 17 00:00:00 2001
From: liuyihang <1905527319@qq.com>
Date: Tue, 10 Sep 2024 06:26:58 +0000
Subject: [PATCH] =?UTF-8?q?update=20ResNet50=5FID4149=5Ffor=5FPyTorch/test?=
 =?UTF-8?q?/train=5Fperformance=5F8p.sh.=20=E6=96=B0=E5=A2=9Eprofiling?=
 =?UTF-8?q?=E4=BC=A0=E5=8F=82=EF=BC=8C=E5=8A=9F=E8=83=BD=E6=9C=AC=E5=9C=B0?=
 =?UTF-8?q?=E5=B7=B2=E9=AA=8C=E8=AF=81=EF=BC=8C=E4=B8=8D=E4=BC=A0=E5=8F=82?=
 =?UTF-8?q?=E4=B8=8D=E5=BD=B1=E5=93=8D=E7=94=A8=E4=BE=8B=E6=89=A7=E8=A1=8C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: liuyihang <1905527319@qq.com>
---
 .../ResNet50_ID4149_for_PyTorch/test/train_performance_8p.sh   | 3 +++
 1 file changed, 3 insertions(+)

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 46bf1b47d1..29cd43a97b 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
 
-- 
Gitee