rebuild

configs/test.yml

@@ -0,0 +1,41 @@
+# 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: true                  # 是否启用分布式训练
+
+# 模型配置
+models:
+  backbone: 'resnet18'
+  channel_ratio: 1.0
+  model_path: "./checkpoints/resnet18_scatter_6.2/best.pth"
+  half: false                        # 是否启用半精度测试（fp16）
+
+# 数据配置
+data:
+  group_test: False                 # 数据集名称（示例用，可替换为实际数据集）
+  test_batch_size: 128               # 训练批次大小
+  num_workers: 32                      # 数据加载线程数
+  test_dir: "../data_center/scatter/"   # 验证数据集根目录
+  test_group_json: "../data_center/contrast_learning/model_test_data/test/inner_group_pairs.json"
+  test_list: "../data_center/scatter/val_pair.txt"
+
+transform:
+  img_size: 224                     # 图像尺寸
+  img_mean: 0.5                     # 图像均值
+  img_std: 0.5                      # 图像方差
+  RandomHorizontalFlip: 0.5         # 随机水平翻转概率
+  RandomRotation: 180               # 随机旋转角度
+  ColorJitter: 0.5                   # 随机颜色抖动强度
+
+save:
+  save_dir: ""
+  save_name: ""
+
+