Skip to content

Commit 5711315

Browse files
authored
Update README.md
1 parent 4d92232 commit 5711315

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

README.md

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,8 +90,12 @@ Or you can customize and use ```graph_net.torch.test_compiler``` yourself:
9090
```bash
9191
python -m graph_net.torch.test_compiler \
9292
--model-path $GRAPH_NET_EXTRACT_WORKSPACE/model_name/ \
93-
--compiler /path/to/custom/compiler/ \
93+
--compiler /custom/or/builtin/compiler/ \
94+
--warmup /times/to/warmup/ \
95+
--trials /times/to/test/ \
96+
--device /device/to/execute/ \
9497
--output-dir /path/to/save/JSON/result/file/
98+
9599
# Note: if --compiler is omitted, PyTorch’s built-in compiler is used by default
96100
```
97101

0 commit comments

Comments
 (0)