ONNX
ytfeng commited on
Commit
057f07e
·
1 Parent(s): 3a3d327

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. yolox.py +2 -2
yolox.py CHANGED
@@ -24,8 +24,8 @@ class YoloX:
24
  return self.__class__.__name__
25
 
26
  def setBackendAndTarget(self, backendId, targetId):
27
- self._backendId = backendId
28
- self._targetId = targetId
29
  self.net.setPreferableBackend(self.backendId)
30
  self.net.setPreferableTarget(self.targetId)
31
 
 
24
  return self.__class__.__name__
25
 
26
  def setBackendAndTarget(self, backendId, targetId):
27
+ self.backendId = backendId
28
+ self.targetId = targetId
29
  self.net.setPreferableBackend(self.backendId)
30
  self.net.setPreferableTarget(self.targetId)
31