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..8a9920fb01f3e6c545ba6dd00022b505e7d84750 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=1024 \ --learning-rate=${learning_rate} \ --mom=0.9 \ --weight-decay=1.0e-04 \