Newer
Older
TIASJudgeModel / configs / hparam / cv_hp_convnext_cwinv_lr4e-4.yaml
# ハイパーパラメータの比較(ConvNeXt-Tiny・重み inverse・ややあり除外.seed 42): lr = 4.0e-4(それ以外は configs/imbalance/cv_without_mild_cwinv.yaml と同一.選択は内側の検証 macro F1 で行う.TEST_06)
# 臨床データはサーバ上にのみ置く.手元ではこの設定で実行しない.
# 環境変数 TIAS_DATA_ROOT が設定されていれば data.root より優先される.
data:
  root: ~/data/TIASJudgeModel
  manifest: manifest.csv # data.root からの相対パス
  severities: [strong, normal]

split:
  seed: 42
  n_folds: 5

model:
  name: convnext_tiny
  pretrained: true

train:
  image_size: 224
  batch_size: 32
  max_epochs: 40
  patience: 8
  lr: 4.0e-4
  weight_decay: 0.05
  warmup_epochs: 2
  inner_val_ratio: 0.15
  num_workers: 4
  amp: true
  class_weighting: inverse