Newer
Older
TIASJudgeModel / configs / repeat / cv_resnet50_without_mild_s43.yaml
# 5-fold 交差検証(ややあり除外.held_out は全程度で評価する): 計算サーバ上の実データで学習・評価する(SPEC_02「交差検証の実行」).モデル比較用(ResNet-50.モデル名以外は cv_without_mild.yaml と同一).分割 seed 43 の繰り返し(seed 以外は configs/cv_resnet50_without_mild.yaml と同一)
# 臨床データはサーバ上にのみ置く.手元ではこの設定で実行しない.
# 環境変数 TIAS_DATA_ROOT が設定されていれば data.root より優先される.
data:
  root: ~/data/TIASJudgeModel
  manifest: manifest.csv # data.root からの相対パス
  severities: [strong, normal]

split:
  seed: 43
  n_folds: 5

model:
  name: resnet50
  pretrained: true

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