base.yaml 602 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
# config/base.yaml
base:
  # 通用环境变量
  env_vars:
    LD_LIBRARY_PATH: "/data/wkx/benchmark/env-0204/rocblas-install/lib:$LD_LIBRARY_PATH"
  
  # 通用设置
  common:
    result_dir: "./result"
    device_id: 3
    migraphx_driver: "/opt/dtk/bin/migraphx-driver"
    fp16: true
    iterations: 100
    
  # 监控设置
  monitor:
    log_file: "memory_simple.log"
    interval: 1  # 秒
    
  # 要运行的模型列表
  models_to_run:
    # - ppocr-v5-rec
    # - yolov3
    # - ppocr-v5-det
    # - yolov8n
    # - eva-2
    - resnet50
    # - swin-B
    # - vit-L
    # - inception-v3