ONNX
ytfeng commited on
Commit
e9322d8
·
1 Parent(s): 2b4a4e2

Refactor benchmark (#148)

Browse files

* use mean as default for benchmark metric; change result representation;
add --all for benchmarking all configs at a time

* fix comments

* add --model_exclude

* pretty print

* improve benchmark result table header: from band-xpu to xpu-band

* suppress print message

* update benchmark results on CPU-RPI

* add the new benchmark results on the new intel cpu

* fix backend and target setting in benchmark; pre-modify the names of int8 quantized models

* add results on jetson cpu

* add cuda results

* print target and backend when using --all

* add results on Khadas VIM3

* pretty print results

* true pretty print results

* update results in new format

* fix broken backend and target vars

* fix broken backend and target vars

* fix broken backend and target var

* update benchmark results on many devices

* add db results on Ascend-310

* update info on CPU-INTEL

* update usage of the new benchmark script

Files changed (1) hide show
  1. mobilenet.py +2 -2
mobilenet.py CHANGED
@@ -34,8 +34,8 @@ class MobileNet:
34
  return self.__class__.__name__
35
 
36
  def setBackendAndTarget(self, backendId, targetId):
37
- self._backendId = backendId
38
- self._targetId = targetId
39
  self.model.setPreferableBackend(self.backend_id)
40
  self.model.setPreferableTarget(self.target_id)
41
 
 
34
  return self.__class__.__name__
35
 
36
  def setBackendAndTarget(self, backendId, targetId):
37
+ self.backend_id = backendId
38
+ self.target_id = targetId
39
  self.model.setPreferableBackend(self.backend_id)
40
  self.model.setPreferableTarget(self.target_id)
41