From 3a9f7c8abc1953c4fe8a2afcde4de48638c2e59d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E6=B5=AA?= Date: Thu, 24 Mar 2022 08:08:58 +0000 Subject: [PATCH] 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