Skip to content

Commit cc7d1ac

Browse files
author
Zhang Yan
committed
add autoscheduler
1 parent c954e2d commit cc7d1ac

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tune_autoscheduler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -113,8 +113,8 @@ def auto_scheduler_tune(network, batch_size, dtype, target, log_file, inC, outC)
113113
parser.add_argument(
114114
"--network",
115115
type=str,
116-
choices=["MLP1", "MLP2", "MHA1", "MHA2", "MHA3", "MHA4", "single_matmul", "all"],
117-
default="single_matmul",
116+
choices=["MLP1", "MLP2", "MHA1", "MHA2", "MHA3", "MHA4", "all"],
117+
default="MHA4",
118118
help="The name of the neural network.",
119119
)
120120
parser.add_argument("--batch-size", type=int, default=32, help="The batch size")

0 commit comments

Comments
 (0)