Skip to content

T
tic
Commit 32067196 by Alexander Pivovarov Committed by eqy
Options

Fix typo in Evaluate inference time cost code (#2542)

parent 4d8ecb37
Showing with 1 additions and 1 deletions
tutorials/autotvm/tune_nnvm_mobile_gpu.py
......@@ -335,7 +335,7 @@ def tune_and_evaluate(tuning_opt):
# evaluate
print("Evaluate inference time cost...")
ftimer = module.module.time_evaluator("run", ctx, number==1, repeat=30)
ftimer = module.module.time_evaluator("run", ctx, number=1, repeat=30)
prof_res = np.array(ftimer().results) * 1000 # convert to millisecond
print("Mean inference time (std dev): %.2f ms (%.2f ms)" %
(np.mean(prof_res), np.std(prof_res)))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment