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

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

# 模型配置
models:
  backbone: 'resnet18'
  channel_ratio: 0.75
  checkpoints: "../checkpoints/resnet18_1009/best.pth"

# 数据配置
data:
  train_batch_size: 128               # 训练批次大小
  test_batch_size: 128                 # 验证批次大小
  num_workers: 32                      # 数据加载线程数
  half:  true                         # 是否启用半精度数据
  img_dirs_path: "/shareData/temp_data/comparison/Hangzhou_Yunhe/base_data/05-09"
#  img_dirs_path: "/home/lc/contrast_nettest/data/feature_json"
  xlsx_pth: false  # 过滤商品， 默认None不进行过滤

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

# 日志与监控
logging:
  logging_dir: "./logs"                  # 日志保存目录
  tensorboard: true                   # 是否启用TensorBoard
  checkpoint_interval: 30             # 检查点保存间隔（epoch）

save:
  json_bin: "../search_library/yunhedian_05-09.json"       # 保存整个json文件
  json_path: "../data/feature_json_compare/"    # 保存单个json文件
  error_barcodes: "error_barcodes.txt"
  barcodes_statistics: "../search_library/barcodes_statistics.txt"