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)