EMCAD/configs/default.yaml

# EMCAD 默认配置文件

# 数据集配置
dataset:
  name: "ACDC"  # Synapse 或 ACDC
  root_path: "/data/ACDC/train"
  volume_path: "/data/ACDC/test"
  list_dir: "/data/ACDC/lists_ACDC"
  num_classes: 4
  z_spacing: 1
  img_size: 224

# 模型配置
model:
  encoder: "pvt_v2_b2"  # pvt_v2_b0, pvt_v2_b1, pvt_v2_b2, resnet18, etc.
  num_classes: 4
  kernel_sizes: [1, 3, 5]  # 多尺度卷积核大小
  expansion_factor: 2  # MSCB 模块扩展因子
  dw_parallel: true  # 深度卷积并行模式
  add: true  # 特征图相加模式（False 为拼接模式）
  lgag_ks: 3  # LGAG 模块卷积核大小
  activation: "relu"  # 激活函数: relu 或 relu6
  pretrain: true  # 是否加载预训练权重
  pretrained_dir: "./model_pth/"

# 训练配置
training:
  # 六、避免硬编码超参
  max_epochs: 300
  max_iterations: 50000
  batch_size: 6
  base_lr: 0.0001  # 学习率
  weight_decay: 0.0001
  seed: 2222
  deterministic: true
  n_gpu: 1

  # 损失函数权重
  loss:
    w_ce: 0.3
    w_dice: 0.7

  # 监督模式
  supervision: "mutation"  # mutation, deep_supervision, last_layer

  # 早停配置
  early_stopping:
    enabled: false
    patience: 50
    min_delta: 0.001

  # 学习率调度
  lr_scheduler:
    type: "CosineAnnealingLR"
    T_max: 300
    eta_min: 1e-6

# 输出配置
output:
  snapshot_path: "./experiments/"
  save_interval: 50  # 保存模型的间隔 epoch 数
  log_interval: 50   # 日志输出间隔

# 日志配置
logging:
  level: "INFO"
  format: "[%(asctime)s.%(msecs)03d] %(message)s"
  log_file: "train.log"