Skip to content

Commit

Permalink
Use DALI as default in resnet50 example (#5808)
Browse files Browse the repository at this point in the history
Signed-off-by: Joaquin Anton Guirao <[email protected]>
  • Loading branch information
jantonguirao authored Feb 4, 2025
1 parent e3e7c72 commit c533304
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/examples/use_cases/pytorch/resnet50/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ def parse():

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

0 comments on commit c533304

Please sign in to comment.