configs/config.yaml

augmentation:
  blur: true
  brightness: 0.2
  contrast: 0.2
  horizontal_flip: true
  hue: 0.1
  noise: true
  rotation: 15
  saturation: 0.2
  vertical_flip: true
data:
  batch_size: 32
  class_names:
  - 无病变
  - 轻度
  - 中度
  - 重度
  - 增殖性病变
  image_size: 224
  num_workers: 0
  test_dir: data/aptos2019/processed/test
  train_dir: data/aptos2019/processed/train
  use_class_weights: true
  val_dir: data/aptos2019/processed/val
device:
  gpu_id: 0
  mixed_precision: false
  use_gpu: false
inference:
  grad_cam: true
  model_path: weights/best_model.pth
  threshold: 0.5
  tta: false
logging:
  log_dir: logs
  save_frequency: 10
  tensorboard_dir: runs
model:
  dropout: 0.2
  name: efficientnet_b0
  num_classes: 5
  pretrained: true
optimizer:
  beta1: 0.9
  beta2: 0.999
  momentum: 0.9
  name: adam
training:
  class_weights:
  - 0.7333333492279053
  - 2.4763286113739014
  - 0.4058590531349182
  - 1.9791505336761475
  - 3.797037124633789
  distill: false
  distill_alpha: 0.7
  distill_beta: 0.3
  distill_student: mobilenetv3_small
  distill_teacher: efficientnet_b3
  early_stopping_patience: 10
  epochs: 100
  label_smoothing: 0.05
  learning_rate: 0.001
  model_save_path: weights/best_model.pth
  qat: false
  qat_backend: fbgemm
  qat_epochs: 10
  qat_export_path: weights/qat_model.onnx
  save_best_only: true
  scheduler: cosine
  warmup_epochs: 5
  weight_decay: 0.0005