Newer
Older
TIASJudgeModel / configs / cv_synthetic.yaml
# 手元確認用: 合成データ(scripts/make_synthetic_dataset.py の出力)で交差検証を通す設定
# CPU で数分以内に終わるよう,小さなモデル・事前学習なし・2 エポックにしている.
# data.root の相対パスはカレントディレクトリ(リポジトリのルート)基準.
data:
  root: data/synthetic
  manifest: manifest.csv # data.root からの相対パス
  severities: [strong, mild, normal]

split:
  seed: 42
  n_folds: 5

model:
  name: resnet18
  pretrained: false

train:
  image_size: 64
  batch_size: 8
  max_epochs: 2
  patience: 8
  lr: 1.0e-3
  weight_decay: 0.05
  warmup_epochs: 1
  inner_val_ratio: 0.15
  num_workers: 0
  amp: true