Skip to content

Commit b4260fe

Browse files
committed
Use DALI as default in resnet50 example
Signed-off-by: Joaquin Anton Guirao <janton@nvidia.com>
1 parent 2a1ed8f commit b4260fe

File tree

1 file changed

+1
-1
lines changed
  • docs/examples/use_cases/pytorch/resnet50

1 file changed

+1
-1
lines changed

docs/examples/use_cases/pytorch/resnet50/main.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def parse():
8787

8888
parser.add_argument('--dali_cpu', action='store_true',
8989
help='Runs CPU based version of DALI pipeline.')
90-
parser.add_argument("--data_loader", default="pytorch",
90+
parser.add_argument("--data_loader", default="dali",
9191
choices=["pytorch", "dali", "dali_proxy"],
9292
help='Select data loader: "pytorch" for native PyTorch data loader, '
9393
'"dali" for DALI data loader, or "dali_proxy" for PyTorch dataloader with DALI proxy preprocessing.')

0 commit comments

Comments
 (0)