Skip to content

Commit 1edad3b

Browse files
committed
update docs to use mobilenet instead of hf_Albert
1 parent 78efee8 commit 1edad3b

File tree

1 file changed

+3
-3
lines changed
  • models/turbine_models/custom_models/torchbench

1 file changed

+3
-3
lines changed

models/turbine_models/custom_models/torchbench/README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -61,10 +61,10 @@ cd ..
6161
python ./export.py --target=gfx942 --device=rocm --compile_to=vmfb --performance --inference --precision=fp16 --float16 --external_weights=safetensors --external_weights_dir=./torchbench_weights/
6262
```
6363

64-
### Example of manual benchmark using export and IREE runtime CLI (hf_Albert)
64+
### Example of manual benchmark using export and IREE runtime CLI (mobilenet_v3_large)
6565

6666
```shell
67-
python ./export.py --target=gfx942 --device=rocm --compile_to=vmfb --performance --inference --precision=fp16 --float16 --external_weights=safetensors --external_weights_dir=./torchbench_weights/ --model_id=hf_Albert
67+
python ./export.py --target=gfx942 --device=rocm --compile_to=vmfb --performance --inference --precision=fp16 --float16 --external_weights=safetensors --external_weights_dir=./torchbench_weights/ --model_id=mobilenet_v3_large
6868

69-
iree-benchmark-module --module=generated/hf_Albert_32_fp16_gfx942.vmfb --input=@generated/hf_Albert_input0.npy --parameters=model=./torchbench_weights/hf_Albert_fp16.irpa --device=hip://0 --device_allocator=caching --function=main --benchmark_repetitions=10
69+
iree-benchmark-module --module=generated/mobilenet_v3_large_256_fp16_gfx942.vmfb --input=@generated/mobilenet_v3_large_input0.npy --parameters=model=./torchbench_weights/mobilenet_v3_large_fp16.irpa --device=hip://0 --device_allocator=caching --function=main --benchmark_repetitions=10
7070
```

0 commit comments

Comments
 (0)