From d0210ee991805c825b46bcb1e5e2b320afa656b8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E8=92=99=E5=A5=87?= <9953698+wang-mengqi22@user.noreply.gitee.com> Date: Thu, 2 Mar 2023 07:07:24 +0000 Subject: [PATCH] update PyTorch/contrib/cv/semantic_segmentation/UNet_for_PyTorch/test/train_performance_1p.sh. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 王蒙奇 <9953698+wang-mengqi22@user.noreply.gitee.com> --- .../UNet_for_PyTorch/test/train_performance_1p.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/contrib/cv/semantic_segmentation/UNet_for_PyTorch/test/train_performance_1p.sh b/PyTorch/contrib/cv/semantic_segmentation/UNet_for_PyTorch/test/train_performance_1p.sh index cbbf205055..faed718e50 100644 --- a/PyTorch/contrib/cv/semantic_segmentation/UNet_for_PyTorch/test/train_performance_1p.sh +++ b/PyTorch/contrib/cv/semantic_segmentation/UNet_for_PyTorch/test/train_performance_1p.sh @@ -3,7 +3,7 @@ ################基础配置参数,需要模型审视修改################## # 必选字段(必须在此处定义的参数): Network batch_size resume RANK_SIZE # 网络名称,同目录名称 -Network="UNet_for_PyTorch" +Network="UNet_ID3752_for_PyTorch" # 训练batch_size batch_size=16 # 训练使用的npu卡数 -- Gitee