projects
/
platform
/
upstream
/
tvm.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
151f3f5
)
Fix tune_relay_cuda.py (#6001)
author
Lianmin Zheng
<lianminzheng@gmail.com>
Tue, 7 Jul 2020 15:18:31 +0000
(08:18 -0700)
committer
GitHub
<noreply@github.com>
Tue, 7 Jul 2020 15:18:31 +0000
(08:18 -0700)
tutorials/autotvm/tune_relay_cuda.py
patch
|
blob
|
history
diff --git
a/tutorials/autotvm/tune_relay_cuda.py
b/tutorials/autotvm/tune_relay_cuda.py
index
a6fe45b
..
6a995af
100644
(file)
--- a/
tutorials/autotvm/tune_relay_cuda.py
+++ b/
tutorials/autotvm/tune_relay_cuda.py
@@
-128,11
+128,7
@@
tuning_option = {
'measure_option': autotvm.measure_option(
builder=autotvm.LocalBuilder(timeout=10),
- #runner=autotvm.LocalRunner(number=20, repeat=3, timeout=4, min_repeat_ms=150),
- runner=autotvm.RPCRunner(
- '1080ti', # change the device key to your key
- '0.0.0.0', 9190,
- number=20, repeat=3, timeout=4, min_repeat_ms=150)
+ runner=autotvm.LocalRunner(number=20, repeat=3, timeout=4, min_repeat_ms=150),
),
}