Skip to content

Commit 3e935c1

Browse files
committed
resolve comments
Signed-off-by: Will Guo <willg@nvidia.com>
1 parent 7c810fd commit 3e935c1

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

modelopt/onnx/quantization/autotune/__main__.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,11 @@
3535
DEFAULT_TIMING_RUNS = 100
3636
MODE_PRESETS = {
3737
"quick": {"schemes_per_region": 30, "warmup_runs": 10, "timing_runs": 50},
38-
"default": {"schemes_per_region": 50, "warmup_runs": 50, "timing_runs": 100},
38+
"default": {
39+
"schemes_per_region": DEFAULT_NUM_SCHEMES,
40+
"warmup_runs": DEFAULT_WARMUP_RUNS,
41+
"timing_runs": DEFAULT_TIMING_RUNS,
42+
},
3943
"extensive": {"schemes_per_region": 200, "warmup_runs": 50, "timing_runs": 200},
4044
}
4145

0 commit comments

Comments
 (0)