# Experiment config

task: training

experiments_dir: ./experiments

experiment_code: R50_00xxx

num_epochs: 10

# Get dataset and dataloaders

data:

  name: IHDataset

  path: ../data/windowed/

  classes: [noIH, IH]

  num_workers: 8

  batch_size: 32

# Initialize model

model:

  name: resnet50

  feature_extract: False #False, finetune the whole model. True, only update new layer params

  use_pretrained: True

  print_model: False

# Get optimizer

optimizer:

  name: SGD

  lr: 0.01

  momentum: 0.9

  weight_decay: 0.001

  print_params: False

# Get criterion

criterion:

  name: CrossEntropyLoss

# Get scheduler

scheduler:

  name: StepLR

  step_size: 2

  gamma: 0.6