inception-v3.yaml 732 Bytes
Newer Older
wangkx1's avatar
init  
wangkx1 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# config/inception-v3.yaml
model:
  name: "inception-v3"
  
  # 模型文件路径
  # model_file: "/data/wkx/benchmark/models/ppocr-v5-rec_model.onnx"
  
  model_files: 
    - path: "../models/inception/inception-v3.onnx"
      batch: 1
    # - path: "../models/inception/inception-v3_b4.onnx"
    #   batch: 4
    - path: "../models/inception/inception-v3_b8.onnx"
      batch: 8
    # - path: "../models/inception/inception-v3_b16.onnx"
    #   batch: 16


  # 测试的batch大小
  batch_sizes: [1, 8]
  
  # 输入配置
  inputs:
    - name: "input"
      shape: [3, 299, 299]  # BATCH会在运行时插入到第一个位置
  
  # 环境变量
  env_vars:
    # MIGRAPHX_ENABLE_NHWC: 1
    
  # 额外参数
  extra_args: []