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 cbbf20505590fade07e9c32be3032e77a0e44067..faed718e5001b825f750fefc25bea305b1bec075 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卡数