We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2a1ed8f commit b4260feCopy full SHA for b4260fe
docs/examples/use_cases/pytorch/resnet50/main.py
@@ -87,7 +87,7 @@ def parse():
87
88
parser.add_argument('--dali_cpu', action='store_true',
89
help='Runs CPU based version of DALI pipeline.')
90
- parser.add_argument("--data_loader", default="pytorch",
+ parser.add_argument("--data_loader", default="dali",
91
choices=["pytorch", "dali", "dali_proxy"],
92
help='Select data loader: "pytorch" for native PyTorch data loader, '
93
'"dali" for DALI data loader, or "dali_proxy" for PyTorch dataloader with DALI proxy preprocessing.')
0 commit comments