ivan lazarevich commited on
Commit
5a6b49d
1 Parent(s): 7107635

add A40 GPU numbers

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. plotting.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ def get_hw_description(hw_name):
10
  'Intel® Core™i7-10875H (CPU, OpenVINO, FP32)': 'https://8074457.fs1.hubspotusercontent-na1.net/hubfs/8074457/YOLOBench%20Hardware%20product%20sheets/Intel_ARK_SpecificationsChart_2023_10_11.pdf',
11
  'Khadas VIM3 (NPU, INT16)': 'https://8074457.fs1.hubspotusercontent-na1.net/hubfs/8074457/YOLOBench%20Hardware%20product%20sheets/khadas_vim3_specs.pdf',
12
  'Orange Pi 5 (NPU, FP16)': 'https://8074457.fs1.hubspotusercontent-na1.net/hubfs/8074457/YOLOBench%20Hardware%20product%20sheets/OrangePi_5_RK3588S_User%20Manual_v1.5.pdf',
 
13
  }
14
 
15
  hw_url = HW_URLS[hw_name]
@@ -98,6 +99,7 @@ with gr.Blocks(theme=gr.themes.Default(secondary_hue=DEEPLITE_DARK_BLUE_GRADIO),
98
  'Intel® Core™i7-10875H (CPU, OpenVINO, FP32)',
99
  'Khadas VIM3 (NPU, INT16)',
100
  'Orange Pi 5 (NPU, FP16)',
 
101
  ],
102
  value='Jetson Nano (GPU, ONNX Runtime, FP32)',
103
  label='Hardware',
 
10
  'Intel® Core™i7-10875H (CPU, OpenVINO, FP32)': 'https://8074457.fs1.hubspotusercontent-na1.net/hubfs/8074457/YOLOBench%20Hardware%20product%20sheets/Intel_ARK_SpecificationsChart_2023_10_11.pdf',
11
  'Khadas VIM3 (NPU, INT16)': 'https://8074457.fs1.hubspotusercontent-na1.net/hubfs/8074457/YOLOBench%20Hardware%20product%20sheets/khadas_vim3_specs.pdf',
12
  'Orange Pi 5 (NPU, FP16)': 'https://8074457.fs1.hubspotusercontent-na1.net/hubfs/8074457/YOLOBench%20Hardware%20product%20sheets/OrangePi_5_RK3588S_User%20Manual_v1.5.pdf',
13
+ 'NVIDIA A40 (GPU, TensorRT, FP32)': 'https://images.nvidia.com/content/Solutions/data-center/a40/nvidia-a40-datasheet.pdf',
14
  }
15
 
16
  hw_url = HW_URLS[hw_name]
 
99
  'Intel® Core™i7-10875H (CPU, OpenVINO, FP32)',
100
  'Khadas VIM3 (NPU, INT16)',
101
  'Orange Pi 5 (NPU, FP16)',
102
+ 'NVIDIA A40 (GPU, TensorRT, FP32)',
103
  ],
104
  value='Jetson Nano (GPU, ONNX Runtime, FP32)',
105
  label='Hardware',
plotting.py CHANGED
@@ -23,6 +23,7 @@ LATENCY_KEYS = {
23
  'Intel® Core™i7-10875H (CPU, OpenVINO, FP32)': 'openvino_latency',
24
  'Khadas VIM3 (NPU, INT16)': 'vim3_latency',
25
  'Orange Pi 5 (NPU, FP16)': 'orange_pi_latency',
 
26
  }
27
 
28
  LATENCY_KEYS_TO_NAMES = {v: k for k, v in LATENCY_KEYS.items()}
 
23
  'Intel® Core™i7-10875H (CPU, OpenVINO, FP32)': 'openvino_latency',
24
  'Khadas VIM3 (NPU, INT16)': 'vim3_latency',
25
  'Orange Pi 5 (NPU, FP16)': 'orange_pi_latency',
26
+ 'NVIDIA A40 (GPU, TensorRT, FP32)': 'a40_trt_latency',
27
  }
28
 
29
  LATENCY_KEYS_TO_NAMES = {v: k for k, v in LATENCY_KEYS.items()}