From e30377834f55ee239e4adcdc1be3813206cd9712 Mon Sep 17 00:00:00 2001 From: Rong He Date: Mon, 6 Jun 2022 06:56:06 +0000 Subject: [PATCH] update PyTorch/contrib/cv/others/SRFlow/test/train_full_8p.sh. --- PyTorch/contrib/cv/others/SRFlow/test/train_full_8p.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/contrib/cv/others/SRFlow/test/train_full_8p.sh b/PyTorch/contrib/cv/others/SRFlow/test/train_full_8p.sh index d21e7a11e8..20cc26d8b5 100644 --- a/PyTorch/contrib/cv/others/SRFlow/test/train_full_8p.sh +++ b/PyTorch/contrib/cv/others/SRFlow/test/train_full_8p.sh @@ -5,7 +5,7 @@ # 网络名称，同目录名称 Network="SRFlow" # 训练batch_size -batch_size=32 +batch_size=24 # 训练使用的npu卡数 export RANK_SIZE=8 # 数据集路径,保持为空,不需要修改 -- Gitee