From b99c4d252c0ba1cb2ce1379597d62f87d6a5691d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E6=B5=AA?= Date: Thu, 24 Mar 2022 07:22:37 +0000 Subject: [PATCH 1/6] update PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_eval_8p.sh. --- .../ResNet34_ID1594_for_PyTorch/test/train_eval_8p.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_eval_8p.sh b/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_eval_8p.sh index 1f28a58241..34a9a9243d 100644 --- a/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_eval_8p.sh +++ b/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_eval_8p.sh @@ -69,7 +69,7 @@ python3.7 ./main.py \ -a resnet34 \ --addr=$(hostname -I |awk '{print $1}') \ --seed=49 \ - --workers=$(nproc) \ + --workers=64 \ --learning-rate=${learning_rate} \ --mom=0.9 \ --weight-decay=1.0e-04 \ -- Gitee From c8e8e8f6cbc097ddb361c54b2a786f93222e06a9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E6=B5=AA?= Date: Thu, 24 Mar 2022 07:23:07 +0000 Subject: [PATCH 2/6] update PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_full_1p.sh. --- .../ResNet34_ID1594_for_PyTorch/test/train_full_1p.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_full_1p.sh b/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_full_1p.sh index a61dce27b7..f7fa935e47 100644 --- a/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_full_1p.sh +++ b/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_full_1p.sh @@ -77,7 +77,7 @@ python3 ./main.py \ -a resnet34 \ --addr=$(hostname -I |awk '{print $1}') \ --seed=49 \ - --workers=$(nproc) \ + --workers=64 \ --learning-rate=${learning_rate} \ --mom=0.9 \ --weight-decay=1.0e-04 \ -- Gitee From 6eb99279e3b118798c70a99efa517eb53e968bdd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E6=B5=AA?= Date: Thu, 24 Mar 2022 07:23:30 +0000 Subject: [PATCH 3/6] update PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_full_8p.sh. --- .../ResNet34_ID1594_for_PyTorch/test/train_full_8p.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_full_8p.sh b/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_full_8p.sh index f7edcb782e..9cfb2b30ed 100644 --- a/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_full_8p.sh +++ b/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_full_8p.sh @@ -65,7 +65,7 @@ python3 ./main.py \ -a resnet34 \ --addr=$(hostname -I |awk '{print $1}') \ --seed=49 \ - --workers=$(nproc) \ + --workers=64 \ --learning-rate=${learning_rate} \ --mom=0.9 \ --weight-decay=1.0e-04 \ -- Gitee From 1378c395538e263b7f0828315908946f47652fbd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E6=B5=AA?= Date: Thu, 24 Mar 2022 07:23:56 +0000 Subject: [PATCH 4/6] update PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_performance_1p.sh. --- .../ResNet34_ID1594_for_PyTorch/test/train_performance_1p.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_performance_1p.sh b/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_performance_1p.sh index 1d7bce633f..1e1829ad45 100644 --- a/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_performance_1p.sh +++ b/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_performance_1p.sh @@ -77,7 +77,7 @@ python3 ./main.py \ -a resnet34 \ --addr=$(hostname -I |awk '{print $1}') \ --seed=49 \ - --workers=$(nproc) \ + --workers=64 \ --learning-rate=${learning_rate} \ --mom=0.9 \ --weight-decay=1.0e-04 \ -- Gitee From affa5d762f33769f9f9f1fb028b0e0c098f7c112 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E6=B5=AA?= Date: Thu, 24 Mar 2022 07:24:24 +0000 Subject: [PATCH 5/6] update PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_performance_8p.sh. --- .../ResNet34_ID1594_for_PyTorch/test/train_performance_8p.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_performance_8p.sh b/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_performance_8p.sh index 2438a17264..a37baa0350 100644 --- a/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_performance_8p.sh +++ b/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/test/train_performance_8p.sh @@ -65,7 +65,7 @@ python3 ./main.py \ -a resnet34 \ --addr=$(hostname -I |awk '{print $1}') \ --seed=49 \ - --workers=$(nproc) \ + --workers=64 \ --learning-rate=${learning_rate} \ --mom=0.9 \ --weight-decay=1.0e-04 \ -- Gitee From a53112082924fa5670c74fe8ee36f3d42df9a15a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E6=B5=AA?= Date: Thu, 24 Mar 2022 07:26:20 +0000 Subject: [PATCH 6/6] update PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/main.py. --- .../cv/classification/ResNet34_ID1594_for_PyTorch/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/main.py b/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/main.py index 57af4f608c..a87679b66e 100644 --- a/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/main.py +++ b/PyTorch/contrib/cv/classification/ResNet34_ID1594_for_PyTorch/main.py @@ -352,7 +352,7 @@ def main_worker(gpu, ngpus_per_node, args): # Data loading code train_loader, train_loader_len, train_sampler = get_pytorch_train_loader(args.data, args.batch_size, - workers=128, distributed=args.distributed) + workers=args.workers, distributed=args.distributed) val_loader = get_pytorch_val_loader(args.data, args.batch_size, args.workers, distributed=False) -- Gitee