# configs/test.yml
# 专为模型训练对比设计的配置文件
# 支持对比不同训练策略（如蒸馏vs独立训练）

# 基础配置
base:
  device: "cuda"                      # 训练设备（cuda/cpu）
  log_level: "info"                   # 日志级别（debug/info/warning/error）
  embedding_size: 256                 # 特征维度
  pin_memory: true                    # 是否启用pin_memory
  distributed: false                  # 是否启用分布式训练   启用热力图时不能用分布式训练

# 模型配置
models:
  backbone: 'resnet18'
  channel_ratio: 1.0
  model_path: "checkpoints/resnet18_electornic_20250806/best.pth"
  #resnet18_20250715_scale=0.75_sub
  #resnet18_20250718_scale=0.75_nosub
  half: false                        # 是否启用半精度测试（fp16）
  contrast_learning: false

# 数据配置
data:
  test_batch_size: 128               # 训练批次大小
  num_workers: 32                      # 数据加载线程数
  test_dir: "../data_center/electornic/v1/val"   # 验证数据集根目录
  test_group_json: "../data_center/contrast_learning/model_test_data/test/inner_group_pairs.json"
  test_list: "../data_center/electornic/v1/cross_same.txt"
  group_test: false
  save_image_joint: false
  image_joint_pth: "./joint_images"

transform:
  img_size: 224                     # 图像尺寸
  img_mean: 0.5                     # 图像均值
  img_std: 0.5                      # 图像方差
  RandomHorizontalFlip: 0.5         # 随机水平翻转概率
  RandomRotation: 180               # 随机旋转角度
  ColorJitter: 0.5                   # 随机颜色抖动强度

heatmap:
    image_joint_pth: "./heatmap_joint_images"
    feature_layer: "layer4"
    show_heatmap: true

save:
  save_dir: ""
  save_name: ""