Diff of /conf/evaluation.yaml [000000] .. [f45789]

Switch to unified view

a b/conf/evaluation.yaml
1
# Experiment config
2
task: evaluation
3
experiments_dir: ./experiments
4
experiment_code: EFFDET_003
5
metrics: False
6
grad_cam: True
7
8
# Get dataset and dataloaders
9
data:
10
  name: IHDataset
11
  path: ../data/windowed/
12
  classes: [noIH, IH]
13
  num_workers: 8
14
  batch_size: 8
15
16
# Initialize model
17
model:
18
  name: efficientdet_d0
19
  feature_extract: False #False, finetune the whole model. True, only update new layer params
20
  use_pretrained: True
21
  print_model: False