vit-L.yaml 637 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
# config/yolov3.yaml
model:
  name: "yolov3"
  
  # 多个模型文件(对应不同batch size)
  model_files:
    - path: "../models/yolov3/yolov3.onnx"
      batch: 1
    - path: "../models/yolov3/yolov3_b4.onnx"
      batch: 4
    - path: "../models/yolov3/yolov3_b8.onnx"
      batch: 8
    - path: "../models/yolov3/yolov3_b16.onnx"
      batch: 16
  
  # 测试的batch大小(如果使用model_files,这里也可以指定)
  batch_sizes: [1, 8]
  
  # 输入配置
  inputs:
    - name: "input"
      shape: [3, 416, 416]
  
  # 环境变量
  env_vars:
    # MIGRAPHX_ENABLE_NHWC: 1
    
  # 额外参数
  extra_args: []