Newer
Older
TIASJudgeModel / configs / hparam / cv_hp_resnet50_lr3e-4.yaml
# ハイパーパラメータの比較(ResNet-50・学習率 1e-3・ややあり除外.seed 42): lr = 3.0e-4(それ以外は configs/lr1e-3/cv_resnet50_without_mild_lr1e-3.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: resnet50
  pretrained: true

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