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 57af4f608c911cc3f4924f6fc55595f78ca4b497..a87679b66e5b5e11694b3bd4b900c0de64d9d43a 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) 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 1f28a582415eb8505c318ff4dd014b6bf0f18ca9..34a9a9243d229bfc5c9588cfaf911596bd601bc1 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 \ 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 a61dce27b7f69ffe0286e9cf7a1e4e51fa60c801..f7fa935e470cab5b2efce001c42cf0e0e260a567 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 \ 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 f7edcb782e0fc4d74bce08b019f68da3c64e6539..9cfb2b30edd07e1b46c48a96af775487b32dcd66 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 \ 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 1d7bce633fc49c4c77a253e3c82953fcb5a6590e..1e1829ad45429a7c605aa0c1277048bbeae3cbc5 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 \ 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 2438a17264ca2b6fbf4ee6538e2cc3ec6279d0b6..a37baa0350e9c5a5f3fae32bd3e2e72a9bbb641a 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 \